Repository for "Hyperbolic Random Forests" (TMLR)

Normalizing flows in PyTorch

Minimal Mamba-2 implementation in PyTorch

YaRN: Efficient Context Window Extension of Large Language Models

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

[ICML 2024 Oral] Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution (https://arxiv.org/abs/2310.16834)

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

Designing bridge trusses with Pytorch autograd

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

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

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

Inference Llama 2 in one file of pure C

Intelligently transform unstructured to structured data

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

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

Stockfish NNUE (Chess evaluation) trainer in Pytorch

Official implementation of Score-based Data Assimilation

A package for conformal prediction with conditional guarantees.

Implementation of Action Matching

kmeans using PyTorch