Alternatives and similar repositories for loop_nest:

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

• bwasti / better_bindings
  Better bindings for Python
  ☆17Updated 2 years ago
• GindaChen / FlexFlashAttention3
  FlexAttention w/ FlashAttention3 Support
  ☆26Updated 7 months ago
• srush / tangent
  Source-to-Source Debuggable Derivatives in Pure Python
  ☆15Updated last year
• NVIDIA / free-threaded-python
  No-GIL Python environment featuring NVIDIA Deep Learning libraries.
  ☆59Updated 3 weeks ago
• nunoplopes / torchy
  A tracing JIT compiler for PyTorch
  ☆13Updated 3 years ago
• iree-org / iree-jax
  ☆51Updated 8 months ago
• facebookresearch / SparseBO
  code associated with paper "Sparse Bayesian Optimization"
  ☆26Updated last year
• facebookresearch / optimizer-plugins
  A demonstration of source code transformation to implement automatic differentiation, compatible with an operation overload style AD libr…
  ☆13Updated 2 years ago
• alexzhang13 / Triton-Puzzles-Solutions
  Personal solutions to the Triton Puzzles
  ☆18Updated 9 months ago
• google-deepmind / einshape
  ☆103Updated 10 months ago
• justindomke / numbat
  NumPy+Jax with named axes and an uncompromising attitude
  ☆20Updated 2 months ago
• google-research / structured-additive-IR
  ☆59Updated this week
• ahennequ / cuda-tensorcores-register-mapping
  ☆18Updated 2 years ago
• facebookresearch / coocmap
  code for paper "Accessing higher dimensions for unsupervised word translation"
  ☆21Updated last year
• nod-ai / transformer-benchmarks
  benchmarking some transformer deployments
  ☆26Updated 2 years ago
• openvinotoolkit / mlas
  ☆11Updated 3 months ago
• manishucsd / py-codegen
  ☆16Updated 7 months ago
• cornellius-gp / linear_operator.old
  A LinearOperator implementation for PyTorch
  ☆18Updated 4 years ago
• facebookresearch / adaptive_scheduling
  Experimental scripts for researching data adaptive learning rate scheduling.
  ☆23Updated last year
• Quansight-Labs / numpy_pytorch_interop
  ☆20Updated last year
• hazan-lab / flash-stu
  PyTorch implementation of the Flash Spectral Transform Unit.
  ☆16Updated 7 months ago
• johnryan465 / pscan
  ☆40Updated last year
• habanero-lab / APPy
  APPy (Annotated Parallelism for Python) enables users to annotate loops and tensor expressions in Python with compiler directives akin to…
  ☆23Updated last week
• cupy / cupy-benchmark
  CuPy Benchmark
  ☆12Updated 6 years ago
• graphcore-research / jax-experimental
  JAX for Graphcore IPU (experimental)
  ☆21Updated last year
• ptillet / triton-llvm-releases
  ☆21Updated last year
• StoneY1 / Reproducing-BowNet
  ☆9Updated 4 years ago
• UmerHA / triton_util
  Make triton easier
  ☆47Updated 10 months ago
• kyutai-labs / jax-flash-attn3
  JAX bindings for the flash-attention3 kernels
  ☆11Updated 9 months ago
• srush / learns-dex
  ☆33Updated 4 years ago