bwasti / better_bindings

Related projects ⓘ

Alternatives and complementary repositories for better_bindings

• nunoplopes / torchy
  A tracing JIT compiler for PyTorch
  ☆12Updated 2 years ago
• facebookresearch / loop_nest
  Loop Nest - Linear algebra compiler and code generator.
  ☆22Updated 2 years ago
• Quansight / pytest-pytorch
  pytest plugin for a better developer experience when working with the PyTorch test suite
  ☆43Updated 2 years ago
• mikex86 / tritonc
  Standalone commandline CLI tool for compiling Triton kernels
  ☆15Updated 2 months ago
• outergroup / vexpr
  Write your code as tree-like expressions, then transform it
  ☆21Updated 10 months ago
• ezyang / torchdbg
  PyTorch centric eager mode debugger
  ☆43Updated 4 months ago
• gmarkall / life-of-a-numba-kernel
  Worked example of the process from Python source to CUDA kernel execution with Numba
  ☆36Updated 2 months ago
• UmerHA / triton_util
  Make triton easier
  ☆41Updated 5 months ago
• data-apis / array-api-comparison
  Data and tooling to compare the API surfaces of various array libraries.
  ☆54Updated 5 months ago
• zasdfgbnm / autonvtx
  Automatically insert nvtx ranges to PyTorch models
  ☆17Updated 3 years ago
• cgarciae / einop
  ☆58Updated 2 years ago
• google-deepmind / einshape
  ☆99Updated 4 months ago
• NVIDIA / free-threaded-python
  No-GIL Python environment featuring NVIDIA Deep Learning libraries.
  ☆23Updated this week
• cornellius-gp / linear_operator.old
  A LinearOperator implementation for PyTorch
  ☆18Updated 3 years ago
• nod-ai / transformer-benchmarks
  benchmarking some transformer deployments
  ☆26Updated last year
• AMYPAD / CuVec
  Unifying Python/C++/CUDA memory: Python buffered array ↔️ `std::vector` ↔️ CUDA managed memory
  ☆81Updated this week
• ezyang / tlparse
  TORCH_LOGS parser for PT2
  ☆22Updated this week
• facebookresearch / loop_tool
  A thin, highly portable toolkit for efficiently compiling dense loop-based computation.
  ☆148Updated last year
• andyljones / modulepickle
  Extends cloudpickle to dynamically pickle modules from your working directory
  ☆10Updated 5 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.
  ☆35Updated 4 months ago
• catid / dataloader
  High-performance tokenized language data-loader for Python C++ extension
  ☆12Updated 4 months ago
• iree-org / iree-jax
  ☆48Updated 3 months ago
• lianakoleva / no-libtorch-compile
  ☆17Updated last month
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆98Updated 2 months ago
• dorjeduck / momograd
  A Learning Journey: Micrograd in Mojo 🔥
  ☆57Updated last month
• Artur-Galstyan / statedict2pytree
  ☆40Updated 4 months ago
• zdevito / custom_loader
  ☆12Updated 3 years ago
• nathanbreitsch / torchmocks
  Test pytorch code with minimal computational overhead
  ☆25Updated last year
• drisspg / transformer_nuggets
  A place to store reusable transformer components of my own creation or found on the interwebs
  ☆44Updated 2 weeks ago