facebookresearch / loop_nest

Related projects ⓘ

Alternatives and complementary repositories for loop_nest

• facebookresearch / adaptive_scheduling
  Experimental scripts for researching data adaptive learning rate scheduling.
  ☆23Updated last year
• iree-org / iree-jax
  ☆48Updated 3 months ago
• bwasti / better_bindings
  Better bindings for Python
  ☆17Updated last year
• manishucsd / py-codegen
  ☆14Updated last month
• 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
• GindaChen / FlexFlashAttention3
  FlexAttention w/ FlashAttention3 Support
  ☆27Updated last month
• BobMcDear / flaim
  Flax Image Models - State-of-the-art pre-trained vision backbones for Flax.
  ☆17Updated last year
• NVIDIA / free-threaded-python
  No-GIL Python environment featuring NVIDIA Deep Learning libraries.
  ☆22Updated this week
• facebookresearch / coocmap
  code for paper "Accessing higher dimensions for unsupervised word translation"
  ☆21Updated last year
• facebookresearch / GCD
  Computing the greatest common divisor with transformers, source code for the paper https//arxiv.org/abs/2308.15594
  ☆12Updated 7 months ago
• FrancescoSaverioZuppichini / pytorch-2.0-benchmark
  Benchmarking PyTorch 2.0 different models
  ☆21Updated last year
• edwardjhu / improved_wasserstein
  Code for our ICLR Trustworthy ML 2020 workshop paper "Improved Image Wasserstein Attacks and Defenses"
  ☆13Updated 4 years ago
• enp1s0 / ozIMMU
  FP64 equivalent GEMM via Int8 Tensor Cores using the Ozaki scheme
  ☆46Updated 2 months ago
• facebookresearch / SparseBO
  code associated with paper "Sparse Bayesian Optimization"
  ☆25Updated last year
• ahennequ / cuda-tensorcores-register-mapping
  ☆18Updated 2 years ago
• nunoplopes / torchy
  A tracing JIT compiler for PyTorch
  ☆12Updated 2 years ago
• StoneY1 / Reproducing-BowNet
  ☆9Updated 3 years ago
• unixpickle / vae-textures
  Texture mapping with variational auto-encoders
  ☆40Updated 3 years ago
• lucidrains / holodeck-pytorch
  Implementation of a holodeck, written in Pytorch
  ☆17Updated last year
• ptillet / triton-llvm-releases
  ☆20Updated last year
• nod-ai / transformer-benchmarks
  benchmarking some transformer deployments
  ☆26Updated last year
• srush / drop7
  ☆18Updated 7 months ago
• f-dangel / einconv
  Convolutions and more as einsum for PyTorch
  ☆13Updated 5 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…
  ☆20Updated last week
• facebookresearch / RLCompOpt
  Learning Compiler Pass Orders using Coreset and Normalized Value Prediction. (ICML 2023)
  ☆18Updated last year
• alexzhang13 / Triton-Puzzles-Solutions
  Personal solutions to the Triton Puzzles
  ☆16Updated 4 months ago
• facebookresearch / baspacho
  Direct solver for sparse SPD matrices for nonlinear optimization. Implements supernodal Cholesky decomposition algorithm, and supports GP…
  ☆85Updated last year