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
  ☆26Updated 6 months ago
• bwasti / better_bindings
  Better bindings for Python
  ☆17Updated 2 years ago
• NVIDIA / free-threaded-python
  No-GIL Python environment featuring NVIDIA Deep Learning libraries.
  ☆53Updated this week
• nunoplopes / torchy
  A tracing JIT compiler for PyTorch
  ☆13Updated 3 years ago
• kyutai-labs / jax-flash-attn3
  JAX bindings for the flash-attention3 kernels
  ☆11Updated 8 months ago
• facebookresearch / adaptive_scheduling
  Experimental scripts for researching data adaptive learning rate scheduling.
  ☆23Updated last year
• iree-org / iree-jax
  ☆51Updated 8 months ago
• facebookresearch / SparseBO
  code associated with paper "Sparse Bayesian Optimization"
  ☆26Updated last year
• lucidrains / firefly-torch
  Exploration into the Firefly algorithm in Pytorch
  ☆37Updated 2 months ago
• srush / drop7
  ☆18Updated 11 months ago
• google-research / structured-additive-IR
  ☆58Updated this week
• facebookresearch / GCD
  Computing the greatest common divisor with transformers, source code for the paper https//arxiv.org/abs/2308.15594
  ☆15Updated last year
• facebookresearch / coocmap
  code for paper "Accessing higher dimensions for unsupervised word translation"
  ☆21Updated last year
• manya-bansal / mosaic
  ☆14Updated last year
• manishucsd / py-codegen
  ☆16Updated 6 months ago
• zdevito / custom_loader
  ☆13Updated 3 years ago
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆109Updated last week
• srush / learns-dex
  ☆33Updated 4 years ago
• ahennequ / cuda-tensorcores-register-mapping
  ☆18Updated 2 years ago
• 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
• facebookresearch / loop_tool
  A thin, highly portable toolkit for efficiently compiling dense loop-based computation.
  ☆148Updated 2 years ago
• alexzhang13 / Triton-Puzzles-Solutions
  Personal solutions to the Triton Puzzles
  ☆18Updated 8 months ago
• justindomke / numbat
  NumPy+Jax with named axes and an uncompromising attitude
  ☆20Updated last month
• hazan-lab / flash-stu
  PyTorch implementation of the Flash Spectral Transform Unit.
  ☆16Updated 6 months ago
• Jokeren / triton-samples
  ☆27Updated 3 months ago
• 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
• edwardjhu / improved_wasserstein
  Code for our ICLR Trustworthy ML 2020 workshop paper "Improved Image Wasserstein Attacks and Defenses"
  ☆14Updated 4 years ago
• StoneY1 / Reproducing-BowNet
  ☆9Updated 4 years ago
• cornellius-gp / linear_operator.old
  A LinearOperator implementation for PyTorch
  ☆18Updated 4 years ago
• Chillee / CS344_2021
  Udacity CS344 Introduction to Parallell Programming (https://classroom.udacity.com/courses/cs344), with assignments/materials updated to …
  ☆46Updated 3 years ago