Alternatives and similar repositories for loop_nest

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

• GindaChen / FlexFlashAttention3
  FlexAttention w/ FlashAttention3 Support
  ☆27Updated last year
• facebookresearch / adaptive_scheduling
  Experimental scripts for researching data adaptive learning rate scheduling.
  ☆22Updated 2 years ago
• Chillee / CS344_2021
  Udacity CS344 Introduction to Parallell Programming (https://classroom.udacity.com/courses/cs344), with assignments/materials updated to …
  ☆46Updated 4 years ago
• ahennequ / cuda-tensorcores-register-mapping
  ☆19Updated 3 years ago
• manishucsd / py-codegen
  ☆16Updated last year
• NVIDIA / free-threaded-python
  No-GIL Python environment featuring NVIDIA Deep Learning libraries.
  ☆69Updated 8 months ago
• iree-org / iree-jax
  ☆54Updated last year
• facebookresearch / SparseBO
  code associated with paper "Sparse Bayesian Optimization"
  ☆26Updated 2 years ago
• DeMoriarty / custom_matmul_kernels
  Customized matrix multiplication kernels
  ☆57Updated 3 years ago
• hpcgarage / cuASR
  cuASR: CUDA Algebra for Semirings
  ☆42Updated 3 years ago
• srush / tangent
  Source-to-Source Debuggable Derivatives in Pure Python
  ☆15Updated last year
• graphcore / poptorch
  PyTorch interface for the IPU
  ☆181Updated 2 years ago
• HicrestLaboratory / SPARTA
  SParse AcceleRation on Tensor Architecture
  ☆17Updated 8 months ago
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆125Updated last week
• PatWie / cuda-design-patterns
  Some CUDA design patterns and a bit of template magic for CUDA
  ☆157Updated 2 years ago
• belindal / state-tracking
  Code and data for paper "(How) do Language Models Track State?"
  ☆21Updated 8 months ago
• nunoplopes / torchy
  A tracing JIT compiler for PyTorch
  ☆13Updated 4 years ago
• nod-ai / transformer-benchmarks
  benchmarking some transformer deployments
  ☆26Updated last week
• lernapparat / torchhacks
  Hacks for PyTorch
  ☆19Updated 2 years ago
• facebookresearch / coocmap
  code for paper "Accessing higher dimensions for unsupervised word translation"
  ☆22Updated 2 years ago
• Timeroot / CLQO
  Solver for Unconstrained Binary Quadratic Optimization (UBQO, BQO, QUBO) and Max 2-SAT, based on semidefinite relaxation with constraint …
  ☆15Updated 2 years ago
• graphcore-research / out-of-the-box-fp8-training
  Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.
  ☆46Updated last year
• lucidrains / autoregressive-linear-attention-cuda
  CUDA implementation of autoregressive linear attention, with all the latest research findings
  ☆46Updated 2 years ago
• habanero-lab / APPy
  APPy (Annotated Parallelism for Python) enables users to annotate loops and tensor expressions in Python with compiler directives akin to…
  ☆27Updated this week
• UmerHA / triton_util
  Make triton easier
  ☆49Updated last year
• drisspg / transformer_nuggets
  A place to store reusable transformer components of my own creation or found on the interwebs
  ☆63Updated 2 weeks ago
• lucidrains / firefly-torch
  Exploration into the Firefly algorithm in Pytorch
  ☆41Updated 10 months ago
• jotaf98 / shareddataset
  A PyTorch Dataset that caches samples in shared memory, accessible globally to all processes
  ☆22Updated 3 years ago
• ptillet / triton-llvm-releases
  ☆22Updated 2 years ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆81Updated last year