CoreNet: A library for training deep neural networks
Tooling for the Common Objects In 3D dataset
A Python and R autograding solution
DeepEP: an efficient expert-parallel communication library
Uncommon Objects in 3D dataset
A library for scientific machine learning & physics-informed learning
Deep learning optimization library making distributed training easy
Official inference library for Mistral models
Python package built to ease deep learning on graph
Redis for humans
Asynchronous parallel SSH client library.
Graph Neural Network Library for PyTorch
Flax is a neural network library for JAX
JAX-based neural network library
tree is a library for working with nested data structures
Python library for defining and optimizing mathematical expressions
Light-weight, flexible, expressive statistical data testing library
A Python toolbox for performing gradient-free optimization
Collection of common code shared among different research projects
Google Toolbox for Mac
A JAX-native LLM Post-Training Library
TorchMultimodal is a PyTorch library
AIMET is a library that provides advanced quantization and compression
A large open dataset + tools to speed up MRI scans using ML
Training and serving large-scale neural networks