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