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
  ☆38Updated 9 months ago
• IntelLabs / EquiTriton
  EquiTriton is a project that seeks to implement high-performance kernels for commonly used building blocks in equivariant neural networks…
  ☆66Updated 3 weeks ago
• salykova / sgemm.cu
  High-Performance SGEMM on CUDA devices
  ☆115Updated 11 months ago
• alexzhang13 / Triton-Puzzles-Solutions
  Personal solutions to the Triton Puzzles
  ☆20Updated last year
• axonn-ai / axonn
  Parallel framework for training and fine-tuning deep neural networks
  ☆69Updated 2 months ago
• wiegerw / nerva
  C++ and Python libraries for neural networks.
  ☆18Updated last month
• spcl / daceml
  A Data-Centric Compiler for Machine Learning
  ☆85Updated 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
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆81Updated last year
• openxla / tokamax
  Tokamax: A GPU and TPU kernel library.
  ☆149Updated 2 weeks ago
• gpu-mode / discord-cluster-manager
  Write a fast kernel and run it on Discord. See how you compare against the best!
  ☆66Updated 3 weeks ago
• iree-org / iree-jax
  ☆55Updated last year
• HazyResearch / train-tk
  train with kittens!
  ☆63Updated last year
• NERSC / sc22-dl-tutorial
  Material for the SC22 Deep Learning at Scale Tutorial
  ☆41Updated 2 years ago
• yixiaoer / einshard
  Einsum-like high-level array sharding API for JAX
  ☆34Updated last year
• zinccat / Awesome-Triton-Kernels
  Collection of kernels written in Triton language
  ☆174Updated 9 months ago
• HazyResearch / flash-fft-conv
  FlashFFTConv: Efficient Convolutions for Long Sequences with Tensor Cores
  ☆338Updated last year
• insuhan / hyper-attn
  ☆83Updated 2 years ago
• gevtushenko / llm.c
  LLM training in simple, raw C/CUDA
  ☆109Updated last year
• jax-ml / ml_dtypes
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.
  ☆323Updated this week
• Vis4SciML / Landscaper
  Landscaper is a comprehensive Python framework designed for exploring the loss landscapes of deep learning models.
  ☆20Updated last month
• mcbal / spin-model-transformers
  Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX
  ☆46Updated 2 years ago
• daniel-geon-park / triton_bwd
  Automatic differentiation for Triton Kernels
  ☆29Updated 4 months ago
• Jokeren / triton-samples
  ☆28Updated 11 months ago
• yixiaoer / mistral-v0.2-jax
  JAX implementation of the Mistral 7b v0.2 model
  ☆35Updated last year
• fishmingyu / GeoT
  GeoT: Tensor Centric Library for Graph Neural Network via Efficient Segment Reduction on GPU
  ☆24Updated 9 months ago
• shikaiqiu / compute-better-spent
  ☆62Updated last year
• facebookresearch / MODel_opt
  Memory Optimizations for Deep Learning (ICML 2023)
  ☆114Updated last year
• srush / anynp
  Proof-of-concept of global switching between numpy/jax/pytorch in a library.
  ☆18Updated last year
• NERSC / sc24-dl-tutorial
  SC24 Deep Learning at Scale Tutorial Material
  ☆33Updated 11 months ago