andylolu2 Goto Github PK

adaptive-generation

Adaptive-compute generative modeling. Combining PonderNet and VAE/Diffusion models for better computation efficiency. Undergraduate dissertation @Cambridge.

ads-assignment

alpha-reversi

This model implements the Alpha Zero paper on the game of Reversi/Orthello.

anthropic-hackathon

cuda-mnist

Training MLP on MNIST in 2 seconds with pure CUDA

cutlass

CUDA Templates for Linear Algebra Subroutines

datasets

TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...

dotfiles

My .dotfiles for Fish shell

gpt-neox

An implementation of model parallel autoregressive transformers on GPUs, based on the DeepSpeed library.

idiots

ivy

The Unified Machine Learning Framework

jax-diffusion

Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.

jax-logging

Imperative-like logging for JAX

jax-monads

Playing around with monads in JAX

jax-vqvae-gpt

Implementation of VQ-VAE with a GPT-style sampler in the JAX and Haiku ecosystem.

learn-cuda

Learning how to write CUDA

llm-ol

Building a multilingual machine ontologist with language models

ml-weight-decay

Code for exploring effects of weight decay on neural network generalization

nanogpt

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

nanotron

Minimalistic large language model 3D-parallelism training

pi-connect

prolog-tic-tac-toe

Optimal Tic-Tac-Toe player implemented in Prolog

score-sde-inverse

Solving all kinds of (linear) inverse problems with diffusion models.

simplegemm

The simplest but fast implementation of matrix multiplication in CUDA.

sparse-dropout

SparseDrop is a simple, structured, and hardware-friendly variant of dropout that can benefit from sparsity on GPUs.

speedtoxify

Fast :speak_no_evil: Detoxify inference with ONNX runtime

text2emoji

Text to emoji translation via DeepDreaming

tic-tac-gpt

Interpreting sequence models of Tic Tac Toe games.

transformer-lens-docker

Base docker image for Mech Intern research

yourmumbot

YourMumBot is a discord bot that makes your mum jokes based on content sent by users.