xla topic

ALBERT model Pretraining and Fine Tuning using TF2.0

kamalkraj

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

inoryy

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 t...

gordicaleksa

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

elixir-nx

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

n2cholas

Enabling PyTorch on XLA Devices (e.g. Google TPU)

pytorch

Julia on TPUs

JuliaGPU

Extending JAX with custom C++ and CUDA code

dfm

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

mpi4jax

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

HomebrewNLP