hyhieu / easy_pybind
☆32Updated 9 months ago
Alternatives and similar repositories for easy_pybind:
Users that are interested in easy_pybind are comparing it to the libraries listed below
- ☆25Updated last year
- ☆76Updated 8 months ago
- Fast and memory efficient PyTorch implementation of the Perceiver with FlashAttention.☆26Updated 4 months ago
- Focused on fast experimentation and simplicity☆70Updated 3 months ago
- ☆53Updated last year
- Load compute kernels from the Hub☆99Updated this week
- FlashRNN - Fast RNN Kernels with I/O Awareness☆76Updated this week
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters☆125Updated 3 months ago
- ☆87Updated last year
- ☆33Updated 2 months ago
- Experiment of using Tangent to autodiff triton☆78Updated last year
- WIP☆93Updated 7 months ago
- Implementation of the proposed Spline-Based Transformer from Disney Research☆87Updated 4 months ago
- ☆33Updated 6 months ago
- FlexAttention w/ FlashAttention3 Support☆26Updated 5 months ago
- JAX bindings for Flash Attention v2☆89Updated 8 months ago
- Jax like function transformation engine but micro, microjax☆30Updated 5 months ago
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆45Updated 8 months ago
- My take on Flow Matching☆41Updated 2 months ago
- ☆152Updated last year
- Solve puzzles. Learn CUDA.☆63Updated last year
- [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule☆143Updated last week
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training☆123Updated 11 months ago
- ☆22Updated 9 months ago
- CUDA implementation of autoregressive linear attention, with all the latest research findings☆44Updated last year
- Mixed precision training from scratch with Tensors and CUDA☆21Updated 10 months ago
- Exploration into the Firefly algorithm in Pytorch☆35Updated last month
- Experimental scripts for researching data adaptive learning rate scheduling.☆23Updated last year
- A basic pure pytorch implementation of flash attention☆16Updated 4 months ago
- ☆49Updated last year