A high-throughput and memory-efficient inference and serving engine for LLMs

AI agents running research on single-GPU nanochat training automatically

General plug-and-play inference library for Recursive Language Models (RLMs), supporting various sandboxes.

The best ChatGPT that $100 can buy.

Run LLMs with MLX

Python logging made (stupidly) simple

A classified list of meta learning papers based on realm.

DomainBed is a suite to test domain generalization algorithms

Hydra is a framework for elegantly configuring complex applications

Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

A lightweight library for PyTorch training tools and utilities

Fast, general, and tested differentiable structured prediction in PyTorch

Code for the NeurIPS 2017 Paper "Prototypical Networks for Few-shot Learning"

Enabling PyTorch on XLA Devices (e.g. Google TPU)

A game theoretic approach to explain the output of any machine learning model.

Handout for the tutorial "Creating publication-quality figures with matplotlib"

TensorFlow-based neural network library

☁️ 🚀 📊 📈 Evaluating state of the art in AI

An End-To-End, Lightweight and Flexible Platform for Game Research

Deal or No Deal? End-to-End Learning for Negotiation Dialogues

Pytorch implementation of "A simple neural network module for relational reasoning" (Relational Networks)

Implementation of a classification framework from the paper Aggregated Residual Transformations for Deep Neural Networks

Referring Expression Datasets API

Summaries and notes on Deep Learning research papers

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

Interactive Image Generation via Generative Adversarial Networks

A toolkit for developing and comparing reinforcement learning algorithms.

Some advanced tricks with Torch7 explained easily

Efficient, reusable RNNs and LSTMs for torch