JAX, Flax & Transformers 🤗

github.comsite

3 days of talks around JAX / Flax, Transformers, large-scale language modeling and other great topics.

from

JAX

added

2026-10-10

likes

0

similar

1. ▲
   JAX GitHub Discussions github.com

   site category #jax link ↗
2. ▲
   Flax NNX github.com

   site category #jax link ↗

   An evolution on Flax by the same team

3. ▲
   Jax in Action manning.com

   site category #jax link ↗

   A hands-on guide to using JAX for deep learning and other mathematically-intensive applications.

4. ▲
   Flax Models github.com

   site category #jax link ↗

   Collection of models and methods implemented in Flax.

5. ▲
   flaxmodels github.com

   tool category #jax link ↗

   Pretrained models for Jax/Flax.

6. ▲
   JAX: Accelerated machine-learning research via composable function transformations in Python | NeurIPS 2019 | Skye… slideslive.com

   site category #jax link ↗

   JAX intro presentation in Program Transformations for Machine Learning workshop.

   

JAX › Videos: “3 days of talks around JAX / Flax, Transformers, large-scale language modeling and other great topics.”