Projects

JAX Mambas
Port of Mamba papers to JAX and Flax's NNX API, including the kernels.

DS 410 Course Project
Project that uses PySpark to perform topic modeling on multiple terebytes of Common Crawl data.

EchoTorch
Contributor to the most popular Echo State Network library for PyTorch

S4 Replication
Replicated the S4 architecture for self learning purposes