GoMLX -- Accelerated ML Libraries for Go
-
Updated
Jun 10, 2024 - Go
GoMLX -- Accelerated ML Libraries for Go
Enabling PyTorch on XLA Devices (e.g. Google TPU)
Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python ⚡
A practical method for training energy-based language models.
S + Autograd + XLA :: S-parameter based frequency domain circuit simulations and optimizations using JAX.
Access the Xspec models and corresponding JAX/XLA ops.
JAX - A curated list of resources https://github.com/google/jax
Mine verus coin on ARM like Pi, Tablet, Mobile & Other.
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.
Official scala pool repository
Presents comprehensive benchmarks of XLA-compatible pre-trained models in Keras.
Modern Graph TensorFlow implementation of Super-Resolution GAN
Tutorial about How to change your slow tensorflow training faster
Example how to train GPT-2 (XLA + AMP), export to SavedModel and serve with Tensorflow Serving
ALBERT model Pretraining and Fine Tuning using TF2.0
Contains materials for my talk "You don't know TensorFlow".
Add a description, image, and links to the xla topic page so that developers can more easily learn about it.
To associate your repository with the xla topic, visit your repo's landing page and select "manage topics."