Alternatives and similar repositories for stk

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

• habanero-lab / APPy

  View on GitHub
  APPy (Annotated Parallelism for Python) enables users to annotate loops and tensor expressions in Python with compiler directives akin to…
  ☆30Jan 28, 2026Updated 2 weeks ago
• shawntan / scattermoe

  View on GitHub
  Triton-based implementation of Sparse Mixture of Experts.
  ☆265Oct 3, 2025Updated 4 months ago
• yikangshen / megablocks

  View on GitHub
  ☆20May 30, 2024Updated last year
• srush / triton-autodiff

  View on GitHub
  Experiment of using Tangent to autodiff triton
  ☆82Jan 22, 2024Updated 2 years ago
• GindaChen / FlexFlashAttention3

  View on GitHub
  FlexAttention w/ FlashAttention3 Support
  ☆27Oct 5, 2024Updated last year
• Doraemonzzz / Awesome-Triton-Resources

  View on GitHub
  Awesome Triton Resources
  ☆39Apr 27, 2025Updated 9 months ago
• illinois-impact / klap

  View on GitHub
  A source-to-source compiler for optimizing CUDA dynamic parallelism by aggregating launches
  ☆15Jun 21, 2019Updated 6 years ago
• TiledTensor / TiledBench

  View on GitHub
  Benchmark tests supporting the TiledCUDA library.
  ☆18Nov 19, 2024Updated last year
• BobMcDear / attorch

  View on GitHub
  A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.
  ☆596Aug 12, 2025Updated 6 months ago
• glassroom / heinsen_attention

  View on GitHub
  Reference implementation of "Softmax Attention with Constant Cost per Token" (Heinsen, 2024)
  ☆24Jun 6, 2024Updated last year
• mgmalek / efficient_cross_entropy

  View on GitHub
  ☆124May 28, 2024Updated last year
• proger / nanokitchen

  View on GitHub
  Parallel Associative Scan for Language Models
  ☆18Jan 8, 2024Updated 2 years ago
• IBM / triton-dejavu

  View on GitHub
  Framework to reduce autotune overhead to zero for well known deployments.
  ☆96Sep 19, 2025Updated 4 months ago
• AlibabaResearch / flash-llm

  View on GitHub
  Flash-LLM: Enabling Cost-Effective and Highly-Efficient Large Generative Model Inference with Unstructured Sparsity
  ☆234Sep 24, 2023Updated 2 years ago
• subho406 / agalite

  View on GitHub
  AGaLiTe: Approximate Gated Linear Transformers for Online Reinforcement Learning (Published in TMLR)
  ☆23Oct 15, 2024Updated last year
• ethansmith2000 / TransformerExperiments

  View on GitHub
  ☆19Dec 4, 2025Updated 2 months ago
• sjelassi / transformers_ssm_copy

  View on GitHub
  ☆35Feb 26, 2024Updated last year
• microsoft / Tutel

  View on GitHub
  Tutel MoE: Optimized Mixture-of-Experts Library, Support GptOss/DeepSeek/Kimi-K2/Qwen3 using FP8/NVFP4/MXFP4
  ☆965Dec 21, 2025Updated last month
• Doraemonzzz / nanoTransNormer

  View on GitHub
  ☆11Oct 11, 2023Updated 2 years ago
• Deep-Learning-Profiling-Tools / triton-samples

  View on GitHub
  ☆14Mar 8, 2025Updated 11 months ago
• Noahs-ARK / PaLM

  View on GitHub
  PyTorch implementation for PaLM: A Hybrid Parser and Language Model.
  ☆10Jan 7, 2020Updated 6 years ago
• rycolab / aflt-f2023

  View on GitHub
  Advanced Formal Language Theory (263-5352-00L; Frühjahr 2023)
  ☆10Feb 21, 2023Updated 2 years ago
• google-research / sputnik

  View on GitHub
  A library of GPU kernels for sparse matrix operations.
  ☆283Nov 24, 2020Updated 5 years ago
• proger / accelerated-scan

  View on GitHub
  Accelerated First Order Parallel Associative Scan
  ☆196Jan 7, 2026Updated last month
• renll / SeqBoat

  View on GitHub
  [NeurIPS 2023] Sparse Modular Activation for Efficient Sequence Modeling
  ☆40Dec 2, 2023Updated 2 years ago
• samblouir / birdie

  View on GitHub
  ☆13Dec 15, 2025Updated 2 months ago
• bdusell / stack-attention

  View on GitHub
  Code for the paper "Stack Attention: Improving the Ability of Transformers to Model Hierarchical Patterns"
  ☆18Mar 15, 2024Updated last year
• Aleph-Alpha-Research / NeurIPS-WANT-submission-efficient-parallelization-layouts

  View on GitHub
  ☆22Dec 15, 2023Updated 2 years ago
• JonasGeiping / linear_cross_entropy_loss

  View on GitHub
  A fusion of a linear layer and a cross entropy loss, written for pytorch in triton.
  ☆75Aug 2, 2024Updated last year
• thinking-machines-lab / manifolds

  View on GitHub
  Supporting code for the blog post on modular manifolds.
  ☆115Sep 26, 2025Updated 4 months ago
• OpenNLPLab / ETSC-Exact-Toeplitz-to-SSM-Conversion

  View on GitHub
  [EMNLP 2023] Official implementation of the algorithm ETSC: Exact Toeplitz-to-SSM Conversion our EMNLP 2023 paper - Accelerating Toeplitz…
  ☆14Oct 17, 2023Updated 2 years ago
• jenni-ai / T2FW

  View on GitHub
  Fine-Tuning Pre-trained Transformers into Decaying Fast Weights
  ☆19Oct 9, 2022Updated 3 years ago
• roastduck / FreeTensor

  View on GitHub
  A language and compiler for irregular tensor programs.
  ☆151Nov 29, 2024Updated last year
• LeiWang1999 / tvm_gpu_gemm

  View on GitHub
  play gemm with tvm
  ☆91Jul 22, 2023Updated 2 years ago
• Doraemonzzz / hgru-pytorch

  View on GitHub
  ☆29Jul 9, 2024Updated last year
• xxcclong / GNN-Computing

  View on GitHub
  Artifact for PPoPP20 "Understanding and Bridging the Gaps in Current GNN Performance Optimizations"
  ☆40Nov 16, 2021Updated 4 years ago
• AlirezaMorsali / MLP-Attention

  View on GitHub
  ☆16Dec 19, 2024Updated last year
• gsampler9 / gSampler

  View on GitHub
  ☆28Aug 14, 2024Updated last year
• dangxingyu / rnn-icrag

  View on GitHub
  Official repository of paper "RNNs Are Not Transformers (Yet): The Key Bottleneck on In-context Retrieval"
  ☆27Apr 17, 2024Updated last year