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
• Chillee / CS344_2021
  Udacity CS344 Introduction to Parallell Programming (https://classroom.udacity.com/courses/cs344), with assignments/materials updated to …
  ☆46Updated 4 years ago
• facebookresearch / adaptive_scheduling
  Experimental scripts for researching data adaptive learning rate scheduling.
  ☆22Updated 2 years ago
• iree-org / iree-jax
  ☆55Updated last year
• ahennequ / cuda-tensorcores-register-mapping
  ☆19Updated 3 years ago
• manishucsd / py-codegen
  ☆16Updated last year
• HicrestLaboratory / SPARTA
  SParse AcceleRation on Tensor Architecture
  ☆17Updated 9 months ago
• DeMoriarty / custom_matmul_kernels
  Customized matrix multiplication kernels
  ☆57Updated 3 years ago
• hpcgarage / cuASR
  cuASR: CUDA Algebra for Semirings
  ☆43Updated 3 years ago
• facebookresearch / optimizer-plugins
  A demonstration of source code transformation to implement automatic differentiation, compatible with an operation overload style AD libr…
  ☆13Updated 3 years ago
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆130Updated last month
• facebookresearch / SparseBO
  code associated with paper "Sparse Bayesian Optimization"
  ☆26Updated 2 years ago
• nunoplopes / torchy
  A tracing JIT compiler for PyTorch
  ☆13Updated 4 years ago
• belindal / state-tracking
  Code and data for paper "(How) do Language Models Track State?"
  ☆21Updated 9 months ago
• unixpickle / vae-textures
  Texture mapping with variational auto-encoders
  ☆40Updated 4 years ago
• Quansight-Labs / numpy_pytorch_interop
  ☆20Updated 2 years ago
• google-research / structured-additive-IR
  ☆63Updated last month
• facebookresearch / coocmap
  code for paper "Accessing higher dimensions for unsupervised word translation"
  ☆22Updated 2 years ago
• imbirik / LieStationaryKernels
  Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"
  ☆45Updated 2 years ago
• Linux-cpp-lisp / opt_einsum_fx
  Einsum optimization using opt_einsum and PyTorch FX graph rewriting
  ☆22Updated 3 years ago
• UmerHA / triton_util
  Make triton easier
  ☆50Updated last year
• srush / tangent
  Source-to-Source Debuggable Derivatives in Pure Python
  ☆15Updated last year
• osqp / osqpth
  The differentiable OSQP solver layer for PyTorch
  ☆65Updated 3 years ago
• bwasti / better_bindings
  Better bindings for Python
  ☆19Updated 2 years ago
• justindomke / numbat
  NumPy+Jax with named axes and an uncompromising attitude
  ☆23Updated 10 months ago
• NVIDIA / free-threaded-python
  No-GIL Python environment featuring NVIDIA Deep Learning libraries.
  ☆69Updated 9 months ago
• alexzhang13 / Triton-Puzzles-Solutions
  Personal solutions to the Triton Puzzles
  ☆20Updated last year
• brentyi / jax_dataclasses
  Pytrees + dataclasses ❤️
  ☆71Updated last month
• amirabbasasadi / RockyML
  ⛰️ RockyML - A High-Performance Scientific Computing Framework for Non-smooth Machine Learning Problems
  ☆20Updated 2 years ago
• drisspg / transformer_nuggets
  A place to store reusable transformer components of my own creation or found on the interwebs
  ☆71Updated last week