#Xla

pytorch

xla

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

n2cholas

awesome-jax

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

gomlx

gomlx

GoMLX: An Accelerated Machine Learning Framework For Go

gordicaleksa

get-started-with-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.

LuxDL

Lux.jl

Elegant and Performant Deep Learning

flaport

sax

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