There are 1 repository under xla topic.
JAX - A curated list of resources https://github.com/google/jax
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
Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload
TensorFlow wheels built for latest CUDA/CuDNN and enabled performance flags: SSE, AVX, FMA; XLA
基于tensorflow1.x的预训练模型调用,支持单机多卡、梯度累积,XLA加速,混合精度。可灵活训练、验证、预测。
Presents comprehensive benchmarks of XLA-compatible pre-trained models in Keras.
Official scala pool repository
JAX bindings for the NVIDIA cuDecomp library
Tensorflow2 training code with jit compiling on multi-GPU.
Contains materials for my talk "You don't know TensorFlow".
Provides code to serialize the different models involved in Stable Diffusion as SavedModels and to compile them with XLA.
Example how to train GPT-2 (XLA + AMP), export to SavedModel and serve with Tensorflow Serving
Automated provisioner of a Google Cloud TPU environment for training in PyTorch
Compiling JAX to WebAssembly for exploring client-side machine learning
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.
As the quality of large language models increases, so do our expectations of what they can do. Since the release of OpenAI's GPT-2, text generation capabilities have received attention. And for good reason - these models can be used for summarization, translation, and even real-time learning in some language tasks.
Mine verus coin on ARM like Pi, Tablet, Mobile & Other.
Optimal choice for 🛰 classification problem.
deep learning inference perf analysis
Deep learning solution for Cassava Leaf Disease Classification, a Kaggle's Research Code Competition using Tensorflow.
Woboq codebrowser for Tensorflow v1.8 with XLA Enabled
Tutorial about How to change your slow tensorflow training faster
Enabling PyTorch on XLA Devices (e.g. Google TPU)
Modern Graph TensorFlow implementation of Super-Resolution GAN