Reference implementations of MLPerf™ training benchmarks
A high performance implementation of HDBSCAN clustering
Fault-tolerant, highly scalable GPU orchestration
Functional Machine Learning
Explainability and Interpretability to Develop Reliable ML models
Hummingbird compiles trained ML models into tensor computation
We write your reusable computer vision tools
Build portable, production-ready MLOps pipelines
State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX
A unified framework for scalable computing
A Pythonic framework to simplify AI service building
A unified framework for machine learning with time series
Trainable models and NN optimization tools
Decentralized deep learning in PyTorch. Built to train models
A cross-platform Python library for differentiable programming
The open-source tool for building high-quality datasets
AI Toolkit for Healthcare Imaging
The Triton Inference Server provides an optimized cloud
Deep learning optimization library making distributed training easy
An Open Source implementation of Notebook LM with more flexibility
Machine Learning automation and tracking
A simple but complete full-attention transformer
Book about interpretable machine learning
Build MLOps Pipelines in Minutes
JAX-based neural network library