fishmingyu / GeoT
GeoT: Tensor Centric Library for Graph Neural Network via Efficient Segment Reduction on GPU
☆22Updated 3 weeks ago
Alternatives and similar repositories for GeoT:
Users that are interested in GeoT are comparing it to the libraries listed below
- ☆21Updated last month
- EquiTriton is a project that seeks to implement high-performance kernels for commonly used building blocks in equivariant neural networks…☆61Updated last month
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆45Updated 9 months ago
- Personal solutions to the Triton Puzzles☆18Updated 9 months ago
- PyTorch centric eager mode debugger☆47Updated 4 months ago
- Make triton easier☆47Updated 10 months ago
- Experiment of using Tangent to autodiff triton☆78Updated last year
- Memory Optimizations for Deep Learning (ICML 2023)☆64Updated last year
- JORA: JAX Tensor-Parallel LoRA Library (ACL 2024)☆33Updated last year
- ☆13Updated last month
- Source-to-Source Debuggable Derivatives in Pure Python☆15Updated last year
- WholeGraph - large scale Graph Neural Networks☆105Updated 5 months ago
- Graph neural networks in JAX.☆67Updated 10 months ago
- A place to store reusable transformer components of my own creation or found on the interwebs☆49Updated last week
- The simplest implementation of recent Sparse Attention patterns for efficient LLM inference.☆59Updated 3 months ago
- ☆27Updated 3 months ago
- ☆29Updated this week
- FlexAttention w/ FlashAttention3 Support☆26Updated 6 months ago
- Experimental paper writing linter.☆34Updated 7 months ago
- Benchmarking PyTorch 2.0 different models☆21Updated 2 years ago
- JAX Scalify: end-to-end scaled arithmetics☆16Updated 5 months ago
- An implementation of the Llama architecture, to instruct and delight☆21Updated 3 months ago
- ☆12Updated 2 months ago
- Torch Distributed Experimental☆115Updated 8 months ago
- Code for the note "NF4 Isn't Information Theoretically Optimal (and that's Good)☆18Updated last year
- This is a port of Mistral-7B model in JAX☆32Updated 9 months ago
- Implementation of Hyena Hierarchy in JAX☆10Updated last year
- CUDA implementation of autoregressive linear attention, with all the latest research findings☆44Updated last year
- Utilities for Training Very Large Models☆58Updated 7 months ago
- Triton Implementation of HyperAttention Algorithm☆47Updated last year