dvruette / barrel-rec-pytorchLinks
☆53Updated 2 years ago
Alternatives and similar repositories for barrel-rec-pytorch
Users that are interested in barrel-rec-pytorch are comparing it to the libraries listed below
Sorting:
- ☆82Updated last year
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training☆132Updated last year
- supporting pytorch FSDP for optimizers☆84Updated last year
- ☆92Updated last year
- ☆62Updated last year
- 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☆92Updated last year
- ☆50Updated last year
- Triton Implementation of HyperAttention Algorithm☆48Updated 2 years ago
- Automatically take good care of your preemptible TPUs