A library for easily evaluating machine learning models and datasets
A unified framework for scalable computing
Topic Modelling for Humans
State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX
Reference implementations of MLPerf™ training benchmarks
A high performance implementation of HDBSCAN clustering
Message Passing Neural Networks for Molecule Property Prediction
Explainability and Interpretability to Develop Reliable ML models
Hummingbird compiles trained ML models into tensor computation
We write your reusable computer vision tools
A computer vision closed-loop learning platform
A framework for real-life data science
Fault-tolerant, highly scalable GPU orchestration
Trainable models and NN optimization tools
Decentralized deep learning in PyTorch. Built to train models
A Pythonic framework to simplify AI service building
A unified framework for machine learning with time series
Functional Machine Learning
A Python library for audio data augmentation
A cross-platform Python library for differentiable programming
Build portable, production-ready MLOps pipelines
Book about interpretable machine learning
PyTorch extensions for fast R&D prototyping and Kaggle farming
Spatiotemporal Signal Processing with Neural Machine Learning Models
Data science on data without acquiring a copy