Alternatives and similar repositories for loop_nest

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

• facebookresearch / adaptive_scheduling
  Experimental scripts for researching data adaptive learning rate scheduling.
  ☆22Updated 2 years ago
• GindaChen / FlexFlashAttention3
  FlexAttention w/ FlashAttention3 Support
  ☆27Updated last year
• facebookresearch / SparseBO
  code associated with paper "Sparse Bayesian Optimization"
  ☆26Updated 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
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆122Updated last month
• iree-org / iree-jax
  ☆52Updated last year
• belindal / state-tracking
  Code and data for paper "(How) do Language Models Track State?"
  ☆19Updated 6 months ago
• DeMoriarty / custom_matmul_kernels
  Customized matrix multiplication kernels
  ☆57Updated 3 years ago
• bwasti / better_bindings
  Better bindings for Python
  ☆19Updated 2 years ago
• lucidrains / firefly-torch
  Exploration into the Firefly algorithm in Pytorch
  ☆41Updated 8 months ago
• ahennequ / cuda-tensorcores-register-mapping
  ☆19Updated 3 years ago
• HicrestLaboratory / SPARTA
  SParse AcceleRation on Tensor Architecture
  ☆17Updated 6 months ago
• nunoplopes / torchy
  A tracing JIT compiler for PyTorch
  ☆13Updated 3 years ago
• unixpickle / vae-textures
  Texture mapping with variational auto-encoders
  ☆40Updated 4 years ago
• srush / tangent
  Source-to-Source Debuggable Derivatives in Pure Python
  ☆15Updated last year
• facebookresearch / coocmap
  code for paper "Accessing higher dimensions for unsupervised word translation"
  ☆21Updated 2 years ago
• lucidrains / blackbox-gradient-sensing
  Implementation and explorations into Blackbox Gradient Sensing (BGS), an evolutionary strategies approach proposed in a Google Deepmind p…
  ☆20Updated 3 months ago
• xflash96 / pybind11_package_example
  Example python package with pybind11 cpp extension
  ☆57Updated 4 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.
  ☆45Updated last year
• justindomke / numbat
  NumPy+Jax with named axes and an uncompromising attitude
  ☆22Updated 7 months ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆80Updated last year
• facebookresearch / loop_tool
  A thin, highly portable toolkit for efficiently compiling dense loop-based computation.
  ☆148Updated 2 years ago
• NVIDIA / free-threaded-python
  No-GIL Python environment featuring NVIDIA Deep Learning libraries.
  ☆64Updated 6 months ago
• google-deepmind / dks
  Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…
  ☆74Updated 3 months ago
• cornellius-gp / linear_operator.old
  A LinearOperator implementation for PyTorch
  ☆18Updated 4 years ago
• danieljtait / jax_xla_adventures
  Dive into Jax, Flax, XLA and C++
  ☆32Updated 5 years ago
• drbh / nnli
  🔭 interactively explore `onnx` networks in your CLI.
  ☆26Updated last year
• manishucsd / py-codegen
  ☆16Updated last year
• brentyi / jax_dataclasses
  Pytrees + dataclasses ❤️
  ☆67Updated 6 months ago
• juntang-zhuang / ACProp-Optimizer
  Implementation for ACProp ( Momentum centering and asynchronous update for adaptive gradient methdos, NeurIPS 2021)
  ☆16Updated 4 years ago