Implementation of Imagen, Google's Text-to-Image Neural Network
Agents write python code to call tools and orchestrate other agents
Retrieval and Retrieval-augmented LLMs
A collaboration friendly studio for NeRFs
A coding-free framework built on PyTorch
A cross-platform Python library for differentiable programming
Scalable and user friendly neural forecasting algorithms.
The easiest way to use deep metric learning in your application
Minimal reproduction of OneRec
MLOps tools for managing & orchestrating the ML LifeCycle
Code release for Cut and Learn for Unsupervised Object Detection
High-resolution models for human tasks
PyTorch code and models for VJEPA2 self-supervised learning from video
LLM training code for MosaicML foundation models
Pytorch domain library for recommendation systems
Learn how to develop, deploy and iterate on production-grade ML
Toolkit for conversational AI
Open source framework for deep learning satellite and aerial imagery
Simple and distributed Machine Learning
Towards Human-Sounding Speech
Play couplet with seq2seq model
A fast TTS architecture with conditional flow matching
Rapid, unbiased, reproducible analysis of synaptic events
Sandbox for training deep learning networks
ktrain is a Python library that makes deep learning AI more accessible