Alternatives and similar repositories for fastkron:

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

• spcl / sten
  Sparsity support for PyTorch
  ☆34Updated last month
• IntelLabs / EquiTriton
  EquiTriton is a project that seeks to implement high-performance kernels for commonly used building blocks in equivariant neural networks…
  ☆61Updated last month
• manishucsd / py-codegen
  ☆16Updated 7 months ago
• spcl / daceml
  A Data-Centric Compiler for Machine Learning
  ☆82Updated last year
• cchan / tccl
  extensible collectives library in triton
  ☆85Updated last month
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆78Updated last year
• nullplay / Unified-Convolution-Framework
  ☆10Updated 2 years ago
• manya-bansal / mosaic
  ☆15Updated last year
• north-numerical-computing / tensor-cores-numerical-behavior
  Test suite for probing the numerical behavior of NVIDIA tensor cores
  ☆38Updated 9 months ago
• GindaChen / FlexFlashAttention3
  FlexAttention w/ FlashAttention3 Support
  ☆26Updated 7 months ago
• triton-lang / kernels
  ☆78Updated 6 months ago
• salykova / sgemm.cu
  High-Performance SGEMM on CUDA devices
  ☆90Updated 3 months ago
• insuhan / hyper-attn
  ☆81Updated last year
• zinccat / Awesome-Triton-Kernels
  Collection of kernels written in Triton language
  ☆121Updated last month
• axonn-ai / axonn
  A parallel framework for training deep neural networks
  ☆58Updated last month
• Linux-cpp-lisp / opt_einsum_fx
  Einsum optimization using opt_einsum and PyTorch FX graph rewriting
  ☆21Updated 3 years ago
• SparseLinearAlgebra / cuBool
  Sparse linear Boolean algebra for Nvidia Cuda
  ☆24Updated this week
• iree-org / iree-jax
  ☆51Updated 8 months ago
• Dao-AILab / fast-hadamard-transform
  Fast Hadamard transform in CUDA, with a PyTorch interface
  ☆183Updated 11 months ago
• patrick-kidger / quax
  Multiple dispatch over abstract array types in JAX.
  ☆120Updated this week
• alexzhang13 / Triton-Puzzles-Solutions
  Personal solutions to the Triton Puzzles
  ☆18Updated 9 months ago
• Silent-Zebra / twisted-smc-lm
  ☆28Updated last month
• shikaiqiu / compute-better-spent
  ☆52Updated 7 months ago
• lxxue / prefix_sum
  A PyTorch wrapper of parallel exclusive scan in CUDA
  ☆12Updated last year
• olcf / NVIDIA-tensor-core-examples
  ☆18Updated 5 years ago
• yixiaoer / einshard
  Einsum-like high-level array sharding API for JAX
  ☆34Updated 9 months ago
• gpauloski / kfac-pytorch
  Distributed K-FAC preconditioner for PyTorch
  ☆85Updated this week
• dorpxam / einops-cpp
  C++17 implementation of einops for libtorch - clear and reliable tensor manipulations with einstein-like notation
  ☆10Updated last year
• intel / intel-extension-for-openxla
  ☆47Updated 3 weeks ago
• BallisticLA / RandBLAS
  A header-only C++ library for sketching in randomized linear algebra
  ☆88Updated last month