A library for scientific machine learning & physics-informed learning
Fast and memory-efficient exact attention
Supercharge Your Model Training
Trained models & code to predict toxic comments
The easiest way to use deep metric learning in your application
Open Source Computer Vision Library
Synthetic Data Generation for tabular, relational and time series data
A fast library for AutoML and tuning
YOLOv5 is the world's most loved vision AI
Decomposable Multiscale Mixing for Time Series Forecasting
Play couplet with seq2seq model
Advanced AI Explainability for computer vision
Large-Scale Agentic RL for High-Performance CUDA Kernel Generation
Optax is a gradient processing and optimization library for JAX
An easy-to-use & supercharged open-source experiment tracker
AlphaFold 3 inference pipeline
Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code
Neural Network Compression Framework for enhanced OpenVINO
HunyuanVideo: A Systematic Framework For Large Video Generation Model
HY-Motion model for 3D character animation generation
Bring the notion of Model-as-a-Service to life
An AI for Music Generation
From-scratch PyTorch implementation of Google's TurboQuant
Train multi-step agents for real-world tasks using GRPO
An Open-Source AI Agent Platform for Financial Analysis using LLMs