The Anandkumar Lab at Caltech

Popular repositories Loading

1. MaskDiT MaskDiT Public

   Code for Fast Training of Diffusion Models with Masked Transformers

   Python 430 15

2. DiffStateGrad DiffStateGrad Public

   DiffStateGrad [ICLR 2025]

   Python 25 9

3. M-Adam-Low-precision-training M-Adam-Low-precision-training Public

   XiaoyuanLowPrecisionProjectCode

   Python 7

4. Equireg Equireg Public

   Equivariance Regularized Diffusion Sampling

   Python