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…
  ☆66Updated 2 months ago
• spcl / sten
  Sparsity support for PyTorch
  ☆37Updated 8 months ago
• wiegerw / nerva
  C++ and Python libraries for neural networks.
  ☆18Updated 3 weeks 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
  ☆83Updated 2 years ago
• mcbal / spin-model-transformers
  Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX
  ☆45Updated 2 years ago
• salykova / sgemm.cu
  High-Performance SGEMM on CUDA devices
  ☆113Updated 10 months ago
• formll / dog
  DoG is SGD's Best Friend: A Parameter-Free Dynamic Step Size Schedule
  ☆63Updated 2 years ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆81Updated last year
• yixiaoer / mistral-v0.2-jax
  JAX implementation of the Mistral 7b v0.2 model
  ☆35Updated last year
• gpu-mode / discord-cluster-manager
  Write a fast kernel and run it on Discord. See how you compare against the best!
  ☆64Updated 2 weeks ago
• alexzhang13 / Triton-Puzzles-Solutions
  Personal solutions to the Triton Puzzles
  ☆20Updated last year
• microsoft / folx
  Implementation of Forward Laplacian algorithm in JAX
  ☆92Updated 3 weeks ago
• AndPotap / einsum-search
  ☆33Updated last year
• shikaiqiu / compute-better-spent
  ☆62Updated last year
• NX-AI / flashrnn
  FlashRNN - Fast RNN Kernels with I/O Awareness
  ☆171Updated last month
• srush / anynp
  Proof-of-concept of global switching between numpy/jax/pytorch in a library.
  ☆18Updated last year
• daniel-geon-park / triton_bwd
  Automatic differentiation for Triton Kernels
  ☆29Updated 4 months ago
• zinccat / Awesome-Triton-Kernels
  Collection of kernels written in Triton language
  ☆173Updated 8 months ago
• yixiaoer / einshard
  Einsum-like high-level array sharding API for JAX
  ☆34Updated last year
• axonn-ai / axonn
  Parallel framework for training and fine-tuning deep neural networks
  ☆70Updated last month
• HazyResearch / train-tk
  train with kittens!
  ☆63Updated last year
• machine-discovery / deer
  Parallelizing non-linear sequential models over the sequence length
  ☆56Updated 5 months ago
• patrick-kidger / quax
  Multiple dispatch over abstract array types in JAX.
  ☆136Updated 3 weeks ago
• xiayuqing0622 / flex_head_fa
  Fast and memory-efficient exact attention
  ☆74Updated 9 months ago
• athms / mad-lab
  A MAD laboratory to improve AI architecture designs 🧪
  ☆135Updated 11 months ago
• argonne-lcf / inference-endpoints
  This is a repository with examples to run inference endpoints on various ALCF clusters
  ☆26Updated last month
• pilancilab / caldera
  Compressing Large Language Models using Low Precision and Low Rank Decomposition
  ☆106Updated 3 weeks ago
• lindermanlab / elk
  Scalable and Stable Parallelization of Nonlinear RNNS
  ☆27Updated last month
• jax-ml / ml_dtypes
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.
  ☆319Updated last week