Local Deep Research achieves ~95% on SimpleQA benchmark (tested with GPT-4.1-mini). Supports local and cloud LLMs (Ollama, Google, Anthropic, ...). Searches 10+ sources - arXiv, PubMed, web, and yo…

A generative world for general-purpose robotics & embodied AI learning.

Modeling, training, eval, and inference code for OLMo

LLM training code for Databricks foundation models

🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash attention v2.

Machine Learning Engineering Open Book

Central place for the engineering/scaling WG: documentation, SLURM scripts and logs, compute environment and data.

The nnsight package enables interpreting and manipulating the internals of deep learned models.

Compliance automation framework, focused on SOC2

[CVPR 2024] X-Adapter: Adding Universal Compatibility of Plugins for Upgraded Diffusion Model

The Startup CTO's Handbook, a book covering leadership, management and technical topics for leaders of software engineering teams

A latent text-to-image diffusion model

The source code of our paper "Diffsound: discrete diffusion model for text-to-sound generation"

Dataset of prompts, synthetic AI generated images, and aesthetic ratings.

High-Resolution Image Synthesis with Latent Diffusion Models

A Non-Autoregressive Text-to-Speech (NAR-TTS) framework, including official PyTorch implementation of PortaSpeech (NeurIPS 2021) and DiffSpeech (AAAI 2022)

NeRF (Neural Radiance Fields) and NeRF in the Wild using pytorch-lightning

Instant neural graphics primitives: lightning fast NeRF and more

Plenoxels: Radiance Fields without Neural Networks

Liveblocks gives you the building blocks and infrastructure to enable people and AI to work together inside your app.

Rich is a Python library for rich text and beautiful formatting in the terminal.

A curated list of awesome open source libraries to deploy, monitor, version and scale your machine learning

ROADMAP(Mind Map) and KEYWORD for students those who have interest in learning NLP

Show, Attend, and Tell | a PyTorch Tutorial to Image Captioning

🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.

Public repo for Augmented Multiscale Deep InfoMax representation learning

Implementation of Reinforcement Learning Algorithms. Python, OpenAI Gym, Tensorflow. Exercises and Solutions to accompany Sutton's Book and David Silver's course.

StarCraft II - pysc2 Deep Reinforcement Learning Examples

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch