JAX

Read news on JAX with our app.

Read more in the app

SDFs and the Fast sweeping algorithm in Jax

Show HN: Kinematic Hand Skeleton Optimization in Jax

Craftax: (Crafter and NetHack) RL Environment in Jax

Numpyro: Probabilistic programming with NumPy powered by Jax

Defining Statistical Models in Jax?

Deep Learning with Jax

Jax and Equinox: What are they and why should I bother?

The future of Deep Learning frameworks

Shape Rotation 101: An Intro to Einsum and Jax Transformers

Penzai: JAX research toolkit for building, editing, and visualizing neural nets