01

Python

Conservation Laws for Diffusion Models

Reference implementation for masked, uniform-replacement, and Gaussian noise paths, including Markov, text8, and CIFAR-10 experiments and tools for reproducing the paper’s conservation-law evaluations.

02

Python

Neural Polar Decoders

Training and evaluation code for structured neural polar decoders, with mutual-information estimation and input-distribution or code-rate optimization for channels with and without memory.

03

Python

Neural Polar Decoders for DNA Storage

Experiments for deletion, insertion-deletion-substitution, and multi-read DNA storage channels, with learned decoding built around polar-code structure.

More code

See my GitHub profile for additional research repositories and ongoing open-source work.