kastnerkyle

makeMoE

From scratch implementation of a sparse mixture of experts language model inspired by Andrej Karpathy's makemore :)

conditional-conformal

A package for conformal prediction with conditional guarantees.

consistency_models

Official repo for consistency models.

CP_HMM

Ctrl-G

deform_conv2d_onnx_exporter

DFlowCode

dirichlet-flow-matching

FineMerge

flow-matching

heinsen_sequence

Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)

HoroRF

Repository for "Hyperbolic Random Forests" (TMLR)

jam

Implementation of Action Matching

kmeans_pytorch

kmeans using PyTorch

llama2.c

Inference Llama 2 in one file of pure C

mamba-minimal

Simple, minimal implementation of Mamba in one file of PyTorch.

mamba2-minimal

Minimal Mamba-2 implementation in PyTorch

meta-prompt

A re-implementation of Meta-Prompt in LangChain for building self-improving agents.

metropolis-adjusted-MLA

Code for the experiments in the paper (https://arxiv.org/abs/2312.08823)

nnue-pytorch

Stockfish NNUE (Chess evaluation) trainer in Pytorch

PytorchBridge

Designing bridge trusses with Pytorch autograd

sda

Official implementation of Score-based Data Assimilation

simpleaichat

Python package for easily interfacing with chat apps, with robust features and minimal code complexity.

sparse-vqvae

Experimental implementation for a sparse-dictionary based version of the VQ-VAE2 paper

SPCI-code

Official implementation of the ICML 2023 work "Sequential Predictive Conformal Inference"

split-conformal-nonexchangeable

trex

Intelligently transform unstructured to structured data

VQGAN-LC

yarn

YaRN: Efficient Context Window Extension of Large Language Models

zuko

Normalizing flows in PyTorch