facebookresearch / loop_nest

Related projects ⓘ

Alternatives and complementary repositories for loop_nest

• NVIDIA / free-threaded-python
  No-GIL Python environment featuring NVIDIA Deep Learning libraries.
  ☆20Updated this week
• GindaChen / FlexFlashAttention3
  FlexAttention w/ FlashAttention3 Support
  ☆26Updated last month
• manishucsd / py-codegen
  ☆14Updated last month
• bwasti / better_bindings
  Better bindings for Python
  ☆17Updated last year
• enp1s0 / ozIMMU
  FP64 equivalent GEMM via Int8 Tensor Cores using the Ozaki scheme
  ☆46Updated 2 months ago
• iree-org / iree-jax
  ☆48Updated 3 months ago
• nunoplopes / torchy
  A tracing JIT compiler for PyTorch
  ☆12Updated 2 years ago
• facebookresearch / adaptive_scheduling
  Experimental scripts for researching data adaptive learning rate scheduling.
  ☆23Updated last year
• zasdfgbnm / autonvtx
  Automatically insert nvtx ranges to PyTorch models
  ☆17Updated 3 years ago
• ptillet / triton-llvm-releases
  ☆20Updated last year
• ahennequ / cuda-tensorcores-register-mapping
  ☆18Updated 2 years ago
• ezyang / tlparse
  TORCH_LOGS parser for PT2
  ☆21Updated 3 weeks ago
• habanero-lab / APPy
  APPy (Annotated Parallelism for Python) enables users to annotate loops and tensor expressions in Python with compiler directives akin to…
  ☆20Updated 7 months 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.
  ☆35Updated 3 months ago
• DeMoriarty / custom_matmul_kernels
  Customized matrix multiplication kernels
  ☆53Updated 2 years ago
• nod-ai / transformer-benchmarks
  benchmarking some transformer deployments
  ☆26Updated last year
• StoneY1 / Reproducing-BowNet
  ☆9Updated 3 years ago
• gau-nernst / quantized-training
  Explore training for quantized models
  ☆10Updated 2 weeks ago
• microsoft / torchy
  A tracing JIT for PyTorch
  ☆17Updated 2 years ago
• microsoft / tensorflow-rematerialization
  Implementation of a Tensorflow XLA rematerialization pass
  ☆15Updated 4 years ago
• zdevito / custom_loader
  ☆12Updated 3 years ago
• mikex86 / tritonc
  Standalone commandline CLI tool for compiling Triton kernels
  ☆15Updated last month
• Z-T-WANG / LaProp-Optimizer
  Codes accompanying the paper "LaProp: a Better Way to Combine Momentum with Adaptive Gradient"
  ☆20Updated 4 years ago
• Jokeren / triton-samples
  ☆23Updated 2 months ago
• lianakoleva / no-libtorch-compile
  ☆17Updated 2 weeks ago
• Doraemonzzz / Awesome-Triton-Resources
  Awesome Triton Resources
  ☆18Updated 3 weeks ago
• mlcommons / mobile_models
  MLPerf™ Mobile models
  ☆24Updated 3 weeks ago
• facebookresearch / coocmap
  code for paper "Accessing higher dimensions for unsupervised word translation"
  ☆21Updated last year
• srush / tangent
  Source-to-Source Debuggable Derivatives in Pure Python
  ☆14Updated 9 months ago