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

LLM training in simple, raw C/CUDA

A library for mechanistic interpretability of GPT-style language models

RL Environments in JAX 🌍

A coding-free framework built on PyTorch for reproducible deep learning studies. PyTorch Ecosystem. 🏆26 knowledge distillation methods presented at CVPR, ICLR, ECCV, NeurIPS, ICCV, etc are implemen…

A large-scale benchmark and learning environment.

Monte Carlo tree search in JAX

OpenAssistant is a chat-based assistant that understands tasks, can interact with third-party systems, and retrieve information dynamically to do so.

PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms.

Open source implementation of InstructGPT (not finished)

The simplest, fastest repository for training/finetuning medium-sized GPTs.

A latent text-to-image diffusion model

Repository to track the progress in Natural Language Processing (NLP), including the datasets and the current state-of-the-art for the most common NLP tasks.

Official repo for team Fnet in Sentiment Analysis and Homophobia detection of YouTube comments in Code-Mixed Dravidian Languages

You like pytorch? You like micrograd? You love tinygrad! ❤️