dogear

enter for all results · esc to close

JAX Toolbox

github.comtool

Nightly CI and optimized examples for JAX on NVIDIA GPUs using libraries such as T5x, Paxml, and Transformer Engine.

visit github.com ↗
0
sign in to dogear this
from
JAX
added
2026-10-10
likes
0

similar

  1. JAXopt github.com

    Hardware accelerated (GPU/TPU), batchable and differentiable optimizers in JAX.

    tool link ↗

  2. MaxText github.com

    A simple, performant and scalable Jax LLM written in pure Python/Jax and targeting Google Cloud TPUs.

    tool topic aitopic programming link ↗

  3. SymJAX github.com

    Symbolic CPU/GPU/TPU programming.

    tool topic programming link ↗

  4. Extending JAX with custom C++ and CUDA code by Dan Foreman-Mackey github.com

    Tutorial demonstrating the infrastructure required to provide custom ops in JAX.

    tool link ↗

  5. Pax github.com

    A Jax-based machine learning framework for training large scale models.

    tool topic ai link ↗

  6. Jax in Action manning.com

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

    site topic ai link ↗

JAX › Libraries > New Libraries: “Nightly CI and optimized examples for JAX on NVIDIA GPUs using libraries such as T5x, Paxml, and Transformer Engine.”