The Anandkumar Lab at Caltech

Popular repositories Loading

1. MaskDiT MaskDiTPublic

   Code for Fast Training of Diffusion Models with Masked Transformers

   Python 430 15

2. DiffStateGrad DiffStateGradPublic

   DiffStateGrad [ICLR 2025]

   Python 25 9

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

   XiaoyuanLowPrecisionProjectCode

   Python 7

4. Equireg EquiregPublic

   Equivariance Regularized Diffusion Sampling

   Python