The Anandkumar Lab at Caltech

Popular repositories Loading

1. MaskDiT MaskDiT Public

   Code for Fast Training of Diffusion Models with Masked Transformers

   Python 434 15

2. DiffStateGrad DiffStateGrad Public

   DiffStateGrad [ICLR 2025]

   Python 27 9

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

   XiaoyuanLowPrecisionProjectCode

   Python 9 1

4. Equireg Equireg Public

   Equivariance Regularized Diffusion Sampling

   Python