The best ChatGPT that $100 can buy.

Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimentation and parallelization, and has demonstrated industry lead…

Reference models and tools for Cloud TPUs.

Orbax provides common checkpointing and persistence utilities for JAX users

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

An Open Source Machine Learning Framework for Everyone

Google core libraries for Java

scikit-learn: machine learning in Python

A collection of useful .gitignore templates