A library for differentiable nonlinear optimization
The largest collection of PyTorch image encoders / backbones
Advanced evolutionary computation library built on top of PyTorch
LLM abstractions that aren't obstructions
A library for accelerating Transformer models on NVIDIA GPUs
A batteries-included library for building AI-powered software
Training PyTorch models with differential privacy
Graph Neural Network Library for PyTorch
Deep learning optimization library: makes distributed training easy
DeepMind's software stack for physics-based simulation
HunyuanVideo: A Systematic Framework For Large Video Generation Model
Structured outputs for llms
Clean Jupyter notebooks of outputs, metadata, and empty cells
High-Performance Symbolic Regression in Python and Julia
A comprehensive set of fairness metrics for datasets
Library extending Jupyter notebooks to integrate with Apache TinkerPop
Geometric deep learning extension library for PyTorch
A modular high-level library to train embodied AI agents
Easy-to-use and powerful NLP library with Awesome model zoo
A lightweight vision library for performing large object detection
Best practices on recommendation systems
Scripts for fine-tuning Meta Llama3 with composable FSDP & PEFT method
Export Django monitoring metrics for Prometheus.io
Volcano Engine Reinforcement Learning for LLMs
A PyTorch-based Speech Toolkit