Unified Model Serving Framework
Modular quant framework
Develop software autonomously
Training PyTorch models with differential privacy
Implementation of 'lightweight' GAN, proposed in ICLR 2021
FAIR Chemistry's library of machine learning methods for chemistry
Implementation of Imagen, Google's Text-to-Image Neural Network
Open Source Differentiable Computer Vision Library
A Python framework for accelerated simulation, data generation
PRML algorithms implemented in Python
All Algorithms implemented in Python
A Python and R autograding solution
tree is a library for working with nested data structures
A Python implementation of global optimization with gaussian processes
Pytest plugin for testing notebooks
OpenCL integration for Python, plus shiny features
Ready-to-run Docker images containing Jupyter applications
Standardized Serverless ML Inference Platform on Kubernetes
A hyperparameter optimization framework
Benchmarking synthetic data generation methods
Public CI, Docker images for popular JAX libraries
Sample applications for Google Kubernetes Engine (GKE)
High-quality implementations of standard and SOTA methods
Superduper: Integrate AI models and machine learning workflows
A JAX research toolkit to build, edit, & visualize neural networks