A simple but complete full-attention transformer
TensorFlow-based neural network library
High quality, fast, modular reference implementation of SSD in PyTorch
Doom-based AI research platform for reinforcement learning
A system for quickly generating training data with weak supervision
Uniform Manifold Approximation and Projection
PyTorch version of Stable Baselines
Beta Machine Learning Toolkit
A toolkit to optimize ML models for deployment for Keras & TensorFlow
Determined, deep learning training platform
Time series forecasting with PyTorch
Unified Model Serving Framework
A library for audio and music analysis, feature extraction
A distributed system for embedding-based vector retrieval
A library for accelerating Transformer models on NVIDIA GPUs
Scalable and user friendly neural forecasting algorithms.
Your open-source LLM evaluation toolkit
Training PyTorch models with differential privacy
Test Suites for validating ML models & data
The open-source tool for building high-quality datasets
Making large AI models cheaper, faster and more accessible
The Unified Machine Learning Framework
The easiest way to use deep metric learning in your application
Graph Neural Network Library for PyTorch
Models and examples built with TensorFlow