Optax is a gradient processing and optimization library for JAX
A modular, primitive-first, python-first PyTorch library
A python library for self-supervised learning on images
Explainability and Interpretability to Develop Reliable ML models
Build MLOps Pipelines in Minutes
A fast library for AutoML and tuning
Python package for AutoML on Tabular Data with Feature Engineering
Deep learning library
Your open-source LLM evaluation toolkit
Training PyTorch models with differential privacy
High quality, fast, modular reference implementation of SSD in PyTorch
Solve end to end problems using Llama model family
MiniSom is a minimalistic implementation of the Self Organizing Maps
TimeGPT-1: production ready pre-trained Time Series Foundation Model
A Python package for segmenting geospatial data with the SAM
Machine Learning automation and tracking
ktrain is a Python library that makes deep learning AI more accessible
Implementation of Denoising Diffusion Probabilistic Model in Pytorch
Jittor is a high-performance deep learning framework
TensorFlow-based neural network library
Django friendly finite state machine support
The Unified Machine Learning Framework
Serve machine learning models within a Docker container
Powering Amazon custom machine learning chips
Library to help with training and evaluating neural networks