xla

Multi-dimensional arrays (tensors) and numerical definitions for Elixir

JAX - A curated list of resources https://github.com/google/jax

Julia on TPUs

The purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku. It contains my "Machine Learning with JAX" series of tutorials (YouTube videos and Jupyter Notebooks) as well as the content I found useful while learning about the JAX ecosystem.

ALBERT model Pretraining and Fine Tuning using TF2.0

Extending JAX with custom C++ and CUDA code

Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python ⚡

TensorFlow wheels built for latest CUDA/CuDNN and enabled performance flags: SSE, AVX, FMA; XLA

Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload

基于tensorflow1.x的预训练模型调用，支持单机多卡、梯度累积，XLA加速，混合精度。可灵活训练、验证、预测。

XLA integration of Open Neural Network Exchange (ONNX)

Tensorflow2 training code with jit compiling on multi-GPU.

Official scala pool repository

S + Autograd + XLA :: S-parameter based frequency domain circuit simulations and optimizations based on JAX.

DEPRECATED ⛔️

Automated provisioner of a Google Cloud TPU environment for training in PyTorch

Optimal choice for 🛰 classification problem.

Classification of multilingual dataset trained only on English training data using pre-trained models. Model is trained on TPUs using PyTorch and torch_xla library.

Woboq codebrowser for Tensorflow v1.8 with XLA Enabled

deep learning inference perf analysis

Compiling JAX to WebAssembly for exploring client-side machine learning

Deep learning solution for Cassava Leaf Disease Classification, a Kaggle's Research Code Competition using Tensorflow.