5 projects
estimint
EIR Estimation using Machine learning interventions
mamba2-jax
Pure JAX/Flax NNX implementation of the Mamba2 state-space model with state caching, pretrained weight loading, and time-series forecasting.
mamba3-jax
Pure JAX/Flax NNX implementation of the Mamba3 state-space model with state caching, weight loading, and time-series forecasting.
minte
Malaria Intervention Emulator (MINTe) - Neural network-based malaria scenario predictions
mamba2-triton-guard
Patch mamba2_torch on Apple Silicon to bypass Triton imports.
