NVIDIA / free-threaded-python

Related projects ⓘ

Alternatives and complementary repositories for free-threaded-python

• GindaChen / FlexFlashAttention3
  FlexAttention w/ FlashAttention3 Support
  ☆27Updated last month
• habanero-lab / APPy
  APPy (Annotated Parallelism for Python) enables users to annotate loops and tensor expressions in Python with compiler directives akin to…
  ☆20Updated this week
• facebookresearch / loop_nest
  Loop Nest - Linear algebra compiler and code generator.
  ☆22Updated 2 years ago
• facebookresearch / MODel_opt
  Memory Optimizations for Deep Learning (ICML 2023)
  ☆59Updated 7 months ago
• NVlabs / COAT
  ☆24Updated last week
• manishucsd / py-codegen
  ☆14Updated last month
• sgl-project / tensorrt-demo
  TensorRT LLM Benchmark Configuration
  ☆11Updated 3 months ago
• lernapparat / torchhacks
  Hacks for PyTorch
  ☆17Updated last year
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆72Updated 9 months ago
• lianakoleva / no-libtorch-compile
  ☆17Updated 2 weeks ago
• lucidrains / autoregressive-linear-attention-cuda
  CUDA implementation of autoregressive linear attention, with all the latest research findings
  ☆43Updated last year
• facebookresearch / adaptive_scheduling
  Experimental scripts for researching data adaptive learning rate scheduling.
  ☆23Updated last year
• mlcommons / logging
  MLPerf™ logging library
  ☆30Updated last week
• FrancescoSaverioZuppichini / pytorch-2.0-benchmark
  Benchmarking PyTorch 2.0 different models
  ☆21Updated last year
• chengzeyi / ParaAttention
  [WIP] Context parallel attention that works with torch.compile
  ☆20Updated this week
• 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
• tridao / flash-attention-wheels
  ☆42Updated 11 months ago
• nod-ai / transformer-benchmarks
  benchmarking some transformer deployments
  ☆26Updated 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
• cchan / tccl
  extensible collectives library in triton
  ☆65Updated last month
• NVIDIA / HMM_sample_code
  CUDA 12.2 HMM demos
  ☆17Updated 3 months ago
• K-Wu / pytorch-direct
  Code for Large Graph Convolutional Network Training with GPU-Oriented Data Communication Architecture (accepted by PVLDB).The outdated wr…
  ☆8Updated last year
• gau-nernst / quantized-training
  Explore training for quantized models
  ☆10Updated this week
• srush / tangent
  Source-to-Source Debuggable Derivatives in Pure Python
  ☆14Updated 9 months ago
• pytorch-labs / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆16Updated this week
• iree-org / iree-jax
  ☆48Updated 3 months ago
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆98Updated last month
• ptillet / triton-llvm-releases
  ☆20Updated last year
• sandeepkumar-skb / pytorch_custom_op
  End to End steps for adding custom ops in PyTorch.
  ☆19Updated 4 years ago