TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...

Scenic: A Jax Library for Computer Vision Research and Beyond

JAX-based neural network library

Optax is a gradient processing and optimization library for JAX.

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

Modular Reinforcement Learning (RL) library (implemented in PyTorch, JAX, and NVIDIA Warp) with support for Gymnasium/Gym, NVIDIA Isaac Lab, MuJoCo Playground and other environments

JAX implementation of OpenAI's Whisper model for up to 70x speed-up on TPU.

PIX is an image processing library in JAX, for JAX.

Second Order Optimization and Curvature Estimation with K-FAC in JAX.

AeroJAX: A differentiable, structure-preserving framework for real-time flow simulation, control, and inverse design. Architected for neural operator integration and latent-space acceleration. Built with JAX.

Probabilistic Programming and Nested sampling in JAX

"chanzi" is a simple and user-friendly JAVA SAST tool that utilizes taint analysis technology, includes built-in common vulnerability rules, supports decompile, custom rule, and is compatible with the technology stacks of Servlet&filter, Spring,struts,Dubbo,Thrift, jax-rs,jax-ws,JFinal,Netty,MyBatis,and JSP.