Computational Applied Mathematics & AI Lab

Popular repositories Loading

1. discretax discretax Public

   Discretax is a light weight collection of state space models implemented in JAX ⚡️

   Python 38 6

2. compressm compressm Public

   This is the official repository for the ICLR 2026 paper "The Curious Case of In-Training Compression of State Space Models".

   Python 34

3. LinearAttentionPruning LinearAttentionPruning Public

   This is the official repository for the pre-print "The Key to State Reduction in Linear Attention: A Rank-based Perspective"

   Python 15

4. neuro-lds neuro-lds Public

   This is the official repository for the ICML 2026 paper "Neural Low-Discrepancy Sequences"

   Python 7

5. Quotient_Recurrent_Reasoner_QRR Quotient_Recurrent_Reasoner_QRR Public

   Official Implementation of QRR

   Python 7 2

6. LoopedActor LoopedActor Public

   Official code for "Looped Actor: Depth-Recurrent Reasoning Models for Reinforcement Learning"

   Python 6 1