Alternatives and similar repositories for GeoT

Users that are interested in GeoT are comparing it to the libraries listed below

• IntelLabs / EquiTriton
  EquiTriton is a project that seeks to implement high-performance kernels for commonly used building blocks in equivariant neural networks…
  ☆61Updated 2 months ago
• alexzhang13 / Triton-Puzzles-Solutions
  Personal solutions to the Triton Puzzles
  ☆18Updated 9 months ago
• lianakoleva / no-libtorch-compile
  ☆21Updated 2 months ago
• drisspg / transformer_nuggets
  A place to store reusable transformer components of my own creation or found on the interwebs
  ☆55Updated this week
• rapidsai / cugraph-gnn
  ☆33Updated this week
• graphcore-research / out-of-the-box-fp8-training
  Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.
  ☆45Updated 9 months ago
• pytorch-labs / tritonparse
  TritonParse is a new tool designed to help developers analyze and debug Triton kernels by visualizing the compilation process and source …
  ☆14Updated last week
• ezyang / torchdbg
  PyTorch centric eager mode debugger
  ☆47Updated 5 months ago
• lucidrains / coordinate-descent-attention
  Implementation of an Attention layer where each head can attend to more than just one token, using coordinate descent to pick topk
  ☆46Updated last year
• GindaChen / FlexFlashAttention3
  FlexAttention w/ FlashAttention3 Support
  ☆26Updated 7 months ago
• Jokeren / triton-samples
  ☆27Updated 4 months ago
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆50Updated 9 months ago
• facebookresearch / MODel_opt
  Memory Optimizations for Deep Learning (ICML 2023)
  ☆64Updated last year
• lucidrains / sinkhorn-router-pytorch
  Self contained pytorch implementation of a sinkhorn based router, for mixture of experts or otherwise
  ☆34Updated 8 months ago
• wolfecameron / GIST
  Repository for "GIST: Distributed training for large-scale graph convolutional networks"
  ☆15Updated 2 years ago
• facebookresearch / projUNN
  Fast training of unitary deep network layers from low-rank updates
  ☆28Updated 2 years ago
• PiotrNawrot / nano-sparse-attention
  The simplest implementation of recent Sparse Attention patterns for efficient LLM inference.
  ☆60Updated 3 months ago
• stas00 / ml-ways
  ML/DL Math and Method notes
  ☆60Updated last year
• AndyShih12 / LongHorizonTemperatureScaling
  PyTorch implementation for "Long Horizon Temperature Scaling", ICML 2023
  ☆20Updated last year
• shreyansh26 / Attention-Mask-Patterns
  Using FlexAttention to compute attention with different masking patterns
  ☆43Updated 7 months ago
• lucidrains / autoregressive-linear-attention-cuda
  CUDA implementation of autoregressive linear attention, with all the latest research findings
  ☆44Updated last year
• SamsungSAILMontreal / nino
  Code for "Accelerating Training with Neuron Interaction and Nowcasting Networks" [to appear at ICLR 2025]
  ☆19Updated 2 months ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆78Updated last year
• google-deepmind / asyncdiloco
  ☆43Updated last year
• layer6ai-labs / calo-forest
  A scalable implementation of diffusion and flow-matching with XGBoost models, applied to calorimeter data.
  ☆18Updated 6 months ago
• rapidsai / wholegraph
  WholeGraph - large scale Graph Neural Networks
  ☆105Updated 5 months ago
• UmerHA / triton_util
  Make triton easier
  ☆47Updated 11 months ago
• NX-AI / flashrnn
  FlashRNN - Fast RNN Kernels with I/O Awareness
  ☆86Updated last month
• aniquetahir / JORA
  JORA: JAX Tensor-Parallel LoRA Library (ACL 2024)
  ☆33Updated last year
• Ryu1845 / hyena-jax
  Implementation of Hyena Hierarchy in JAX
  ☆10Updated 2 years ago