Alternatives and similar repositories for fastkron

Users that are interested in fastkron 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…
  ☆62Updated this week
• spcl / sten
  Sparsity support for PyTorch
  ☆36Updated 5 months ago
• alexzhang13 / Triton-Puzzles-Solutions
  Personal solutions to the Triton Puzzles
  ☆20Updated last year
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆80Updated last year
• salykova / sgemm.cu
  High-Performance SGEMM on CUDA devices
  ☆99Updated 7 months ago
• srush / anynp
  Proof-of-concept of global switching between numpy/jax/pytorch in a library.
  ☆18Updated last year
• axonn-ai / axonn
  A parallel framework for training deep neural networks
  ☆63Updated 5 months ago
• zinccat / Awesome-Triton-Kernels
  Collection of kernels written in Triton language
  ☆152Updated 4 months ago
• Lemon-cmd / energy-transformer-graph
  This repository contains the official code for Energy Transformer---an efficient Energy-based Transformer variant for graph classificatio…
  ☆25Updated last year
• insuhan / hyper-attn
  ☆81Updated last year
• nebius / kvax
  A FlashAttention implementation for JAX with support for efficient document mask computation and context parallelism.
  ☆140Updated 4 months ago
• formll / dog
  DoG is SGD's Best Friend: A Parameter-Free Dynamic Step Size Schedule
  ☆63Updated 2 years ago
• AndPotap / einsum-search
  ☆32Updated 10 months ago
• spcl / daceml
  A Data-Centric Compiler for Machine Learning
  ☆84Updated last year
• HazyResearch / flash-fft-conv
  FlashFFTConv: Efficient Convolutions for Long Sequences with Tensor Cores
  ☆328Updated 8 months ago
• mcbal / spin-model-transformers
  Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX
  ☆41Updated last year
• gpu-mode / discord-cluster-manager
  Write a fast kernel and run it on Discord. See how you compare against the best!
  ☆51Updated this week
• cchan / tccl
  extensible collectives library in triton
  ☆88Updated 5 months ago
• shikaiqiu / compute-better-spent
  ☆57Updated 10 months ago
• gevtushenko / llm.c
  LLM training in simple, raw C/CUDA
  ☆104Updated last year
• graphcore-research / unit-scaling
  A library for unit scaling in PyTorch
  ☆130Updated last month
• RichardAragon / HyperDimensionalComputingNeuralNetwork
  ☆15Updated 11 months ago
• NX-AI / flashrnn
  FlashRNN - Fast RNN Kernels with I/O Awareness
  ☆96Updated 2 months ago
• north-numerical-computing / tensor-cores-numerical-behavior
  Test suite for probing the numerical behavior of NVIDIA tensor cores
  ☆41Updated last year
• glassroom / heinsen_sequence
  Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)
  ☆94Updated 8 months ago
• HazyResearch / train-tk
  train with kittens!
  ☆62Updated 10 months ago
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆32Updated last year
• nv-legate / legate-sparse
  Legate Sparse is a Legate library that aims to provide a distributed and accelerated drop-in replacement for the scipy.sparse library on …
  ☆23Updated 3 weeks ago
• lessw2020 / triton_kernels_for_fun_and_profit
  Custom kernels in Triton language for accelerating LLMs
  ☆25Updated last year
• KernelTuner / kernel_tuner_tutorial
  A hands-on introduction to tuning GPU kernels using Kernel Tuner https://github.com/KernelTuner/kernel_tuner/
  ☆32Updated 4 months ago