The absolute trainer to light up AI agents
The most intuitive, flexible, way for researchers to build models
A python library for self-supervised learning on images
A fast TTS architecture with conditional flow matching
Fast3R: Towards 3D Reconstruction of 1000+ Images in One Forward Pass
20+ high-performance LLMs with recipes to pretrain, finetune at scale
Scalable generative AI framework built for researchers and developers
Open-weight, large-scale hybrid-attention reasoning model
Large-language-model & vision-language-model based on Linear Attention
Lightning-Fast RL for LLM Reasoning and Agents. Made Simple & Flexible
Time series forecasting with PyTorch
Superfast AI decision making and processing of multi-modal data
Flower: A Friendly Federated Learning Framework
Uncover insights, surface problems, monitor, and fine tune your LLM
Machine learning metrics for distributed, scalable PyTorch application
Spatiotemporal Signal Processing with Neural Machine Learning Models
Toolkit for conversational AI
Tensor search for humans
Library of self-supervised methods for visual representation
PyTorch Lightning + Hydra. A very user-friendly template
Flash enables you to easily configure and run complex AI recipes
Toolbox of models, callbacks, and datasets for AI/ML researchers
A large open dataset + tools to speed up MRI scans using ML
Large-scale linear classification, regression and ranking in Python
Punctuation restoration production-ready model for Russian language