sdpmas

gfn-lm-tuning

Awesome-GFlowNets

A curated list of resources about generative flow networks (GFlowNets).

dreamerv3

Mastering Diverse Domains through World Models

information-bottleneck

Deep Learning & Information Bottleneck

HEPML-LivingReview

Living Review of Machine Learning for Particle Physics

hivemind

Decentralized deep learning in PyTorch. Built to train models on thousands of volunteers across the world.

category-theory-resources

Resources for learning Category Theory for an enthusiast

LAVIS

LAVIS - A One-stop Library for Language-Vision Intelligence

raspy

An interactive exploration of Transformer programming.

Megatron-LM

Ongoing research training transformer models at scale

CatGNN

Prototype for a Category Theory-based GNN Library

TNP-pytorch

Official implementation of Transformer Neural Processes

gflownet

Generative Flow Networks

lila

Code for "Invariance Learning in Deep Neural Networks with Differentiable Laplace Approximations"

clrs

GNCA

Code for "Learning Graph Cellular Automata" (NeurIPS 2021).

G.pt

Official PyTorch Implementation of "Learning to Learn with Generative Models of Neural Network Checkpoints"

iris

Transformers are Sample-Efficient World Models. ICLR 2023, notable top 5%.

walk_in_the_park

neural-network-cuda

A neural network from scratch in CUDA/C++

torchsort

Fast, differentiable sorting and ranking in PyTorch

in-context-learning

torchdiffeq

Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.

triton

Development repository for the Triton language and compiler

ml-compiler-opt

Infrastructure for Machine Learning Guided Optimization (MLGO) in LLVM.

einops

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

minimal-flask-api

A template for 'production ready' Flask APIs with Flask & Gunicorn. Includes template test and error-handling files.

bolt

10x faster matrix and vector operations

cuda_programming

Code from the "CUDA Crash Course" YouTube series by CoffeeBeforeArch

lvt

The PyTorch implementation of Latent Video Transformer.