Top JAX GitHub Repositories & Tools (2026)

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

jQuery Form Plugin

PennyLane is an open-source quantum software platform for quantum computing, quantum machine learning, and quantum chemistry. Create meaningful quantum algorithms, from inspiration to implementation.

NukeViet CMS is multi Content Management System. NukeViet CMS is the 1st open source content management system in Vietnam. NukeViet was awarded the Vietnam Talent 2011, the Ministry of Education and Training Vietnam officially encouraged to use.

A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit and 4-bit floating point (FP8 and FP4) precision on Hopper, Ada and Blackwell GPUs, to provide better performance with lower memory utilization in both training and inference.

a rust-native Modelica compiler for fast, composable, scientific computing

Library for reading and processing ML training data.

A simple, performant, and scalable Jax LLM!