Vinay Ramasesh's repositories
rnn_dynamics
Experiment code for studying rnn dynamics
journalclubs
Updated notebooks originally written for ML@B journal club
qutrit_teleportation
Simulation of the 5 qutrit teleportation experiment at QNL
bashmarks_ripoff
Implementation of bashmarks, based on the original
BIG-bench
Beyond the Imitation Game collaborative benchmark for measuring and extrapolating the capabilities of language models
caliban
Research workflows made easy, locally and in the Cloud.
card-trick
Website for practicing permutation card trick
continual-learning
PyTorch implementation of various methods for continual learning (XdG, EWC, online EWC, SI, LwF, DGR, DGR+distill, RtF, iCaRL).
Continual-Learning-Benchmark
Evaluate three types of task shifting with popular continual learning algorithms.
data_processing
data processing utils
dynamic_notebooks
Pedagogical blog posts and stuff written in Sam's dynamic notebooks
EKFAC
implementation of the EKFAC optimizer
flax
Flax is a neural network library for JAX that is designed for flexibility.
mathematics_dataset
This dataset code generates mathematical question and answer pairs, from a range of question types at roughly school-level difficulty.
MLNotes
Jupyter notebooks following papers
model_measurements
Measuring various model quantities in PyTorch
pytorch-advspheres
Reproduction of the Adversarial Spheres paper in PyTorch
pytorch-seq2seq
Tutorials on implementing a few sequence-to-sequence (seq2seq) models with PyTorch and TorchText.
pytorch-split-dataset
Split datasets in pytorch, i.e. datasets taking only a subset of labels
pytorch_image_classification
PyTorch implementation of image classification models for CIFAR-10/CIFAR-100/MNIST/FashionMNIST/Kuzushiji-MNIST
pytorch_imagenet
pytorch imagenet with xla and split dataset
RMTCode
Code, notes, tutorials to accompany my learning random matrix theory
tensorboard_composer
Write audio to tensorboard
test_hook_project
Project to test adding pre-build and pre-run hooks to Caliban