kvfrans/jax-diffusion-transformer
Implementation of Diffusion Transformer (DiT) in JAX
PythonApache-2.0
Stargazers
- a-tabazaPrincess Sumaya University for Technology
- ali-robot
- amadeuzou
- andreakiroPrev MIT, IBM, Amazon
- antferdom@datacrunch-research
- cgarciaeGoogle
- cywinskiWarsaw University of Technology
- Davidnet@updata-ca
- EricBoittierUniversity of Basel
- FeFBBrazil
- giantvision
- heyalchang
- homerjedMünchen, Deutschland
- jby1993
- jiangzhengkaiHKUST
- junhua-lShenzhen
- lishuai-97Beijing
- LIVazquezS
- lucasliunju
- lwtwl23Shanghai
- mstoelzleMIT LIDS
- PerceptronVLondon / Hong Kong
- Ryu1845
- sanglinweiTsinghua University
- shareeff
- shobhitagrawal1
- smsharmaAnthropic
- soryxieTongJi University
- soumik12345@wandb
- stevenygdStanford
- tomdps@CEA-List @ENSTA-U2IS
- ubermenchh
- utopic-dev
- Yannlecun
- yaodongyuUC Berkeley
- YuliangXiuMax Planck Institute for Intelligent Systems