A Python package for segmenting geospatial data with the SAM
Visualizer for neural network, deep learning, machine learning models
Topic Modelling for Humans
Decentralized deep learning in PyTorch. Built to train models
Reference implementations of MLPerf™ training benchmarks
AI Toolkit for Healthcare Imaging
Explainability and Interpretability to Develop Reliable ML models
Hummingbird compiles trained ML models into tensor computation
Core ML tools contain supporting tools for Core ML model conversion
We write your reusable computer vision tools
Open deep learning compiler stack for cpu, gpu, etc.
A framework for real-life data science
Functional Machine Learning
A unified framework for scalable computing
Machine Learning automation and tracking
Models for the spaCy Natural Language Processing (NLP) library
Open Source Computer Vision Library
A Pythonic framework to simplify AI service building
A unified framework for machine learning with time series
Uncover insights, surface problems, monitor, and fine tune your LLM
A cross-platform Python library for differentiable programming
A high performance implementation of HDBSCAN clustering
Fault-tolerant, highly scalable GPU orchestration
Book about interpretable machine learning
JAX-based neural network library