greydanus

stable-diffusion

A latent text-to-image diffusion model

nanoGPT

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

tinygrad

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

llm.c

LLM training in simple, raw C/CUDA

vit-pytorch

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

flax

Flax is a neural network library for JAX that is designed for flexibility.

DALI

A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.

minikeyvalue

A distributed key value store in under 1000 lines. Used in production at comma.ai

makemore

An autoregressive character-level language model for making more things

neural-tangents

Fast and Easy Infinite Neural Networks in Python

TensorNetwork

A library for easy and efficient manipulation of tensor networks.

jax-md

Differentiable, Hardware Accelerated, Molecular Dynamics

celluloid

:movie_camera: Matplotlib animations made easy

pytorch_sparse

PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations

evojax

latent-scope

A scientific instrument for investigating latent spaces

lagrangian_nns

Lagrangian Neural Networks

hamiltonian-nn

Code for our paper "Hamiltonian Neural Networks"

blogmaker

The self-made software package I used to make my blog. Feel free to use it!

mnist1d

A 1D analogue of the MNIST dataset for measuring spatial biases and answering Science of Deep Learning questions.

spectral_inference_networks

Implementation of Spectral Inference Networks, ICLR 2019

neural-structural-optimization

Neural reparameterization improves structural optimization

Growing-Neural-Cellular-Automata-Pytorch

Extended experiments of "Growing Neural Cellular Automata" https://distill.pub/2020/growing-ca/

mmn

Moore Machine Networks (MMN): Learning Finite-State Representations of Recurrent Policy Networks

deep-image-prior-tensorflow

an implementation of Deep Image Prior using tensorflow

ncf

Nature's Cost Function (NCF). Finding paths of least action with gradient descent.

cursivetransformer

Training a transformer to generate cursive

CXResnet

Chest X-Ray Image classification using PyTorch.

post--td-paths

The Paths Perspective on Value Learning

unstable-diffusion