Jamie Townsend

craystack. Compression tools for machine learning researchers

vdvae-jax. Very deep VAEs in JAX/Flax

iprofiler. Interactive code profiling widget for IPython + Jupyter Notebook

fastar. Python

scanagram. Tidy autoregressive inference in JAX

flipper. Reversible programming in Agda

ans-notes. Tutorial paper and code for rANS

ssm-code. Code for the paper "Lossless compression with state space models using bits back coding" - https://arxiv.org/abs/2103.10150

crayjax. JAX implementation of vectorized rANS.

j-towns.github.io. Jupyter Notebook

flax. Flax is a neural network library for JAX that is designed for flexibility.

vANS. Formally verified implementation of asymmetric numeral systems (ANS)