We write your reusable computer vision tools
Machine learning in Python
YOLOv5 is the world's most loved vision AI
A very simple framework for state-of-the-art NLP
JAX-based neural network library
A lightweight audio-to-MIDI converter with pitch bend detection
Models for the spaCy Natural Language Processing (NLP) library
A simple but complete full-attention transformer
JAX-based neural network library
A Pythonic framework to simplify AI service building
Modular quant framework
Label Studio is a multi-type data labeling and annotation tool
The official Python client for the Huggingface Hub
Book about interpretable machine learning
MiniSom is a minimalistic implementation of the Self Organizing Maps
Usable Implementation of "Bootstrap Your Own Latent" self-supervised
Library to help with training and evaluating neural networks
Build portable, production-ready MLOps pipelines
Probabilistic time series modeling in Python
Tensor Learning in Python
Machine learning metrics for distributed, scalable PyTorch application
Optax is a gradient processing and optimization library for JAX
TensorFlow-based neural network library
Graph Neural Network Library for PyTorch
SkyPilot: Run AI and batch jobs on any infra