kelechi-c / dit_flowLinks
DiT (training + flow matching) in Jax
☆11Updated last year
Alternatives and similar repositories for dit_flow
Users that are interested in dit_flow are comparing it to the libraries listed below
Sorting:
- JAX Implementation of Black Forest Labs' Flux.1 family of models☆40Updated 2 months ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆92Updated 2 years ago
- Implementation of GateLoop Transformer in Pytorch and Jax