bwasti / better_bindings

Related projects ⓘ

Alternatives and complementary repositories for better_bindings

• facebookresearch / loop_tool
  A thin, highly portable toolkit for efficiently compiling dense loop-based computation.
  ☆147Updated last year
• facebookresearch / loop_nest
  Loop Nest - Linear algebra compiler and code generator.
  ☆22Updated 2 years ago
• nunoplopes / torchy
  A tracing JIT compiler for PyTorch
  ☆12Updated 2 years ago
• outergroup / vexpr
  Write your code as tree-like expressions, then transform it
  ☆21Updated 10 months ago
• nv-legate / legate.pandas
  An Aspiring Drop-In Replacement for Pandas at Scale
  ☆73Updated 3 years ago
• cgarciae / einop
  ☆58Updated 2 years ago
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆58Updated last year
• AMYPAD / CuVec
  Unifying Python/C++/CUDA memory: Python buffered array ↔️ `std::vector` ↔️ CUDA managed memory
  ☆81Updated last week
• UmerHA / triton_util
  Make triton easier
  ☆41Updated 4 months ago
• ezyang / torchdbg
  PyTorch centric eager mode debugger
  ☆43Updated 4 months ago
• data-apis / array-api-comparison
  Data and tooling to compare the API surfaces of various array libraries.
  ☆54Updated 5 months ago
• iree-org / iree-jax
  ☆48Updated 3 months ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆71Updated 9 months ago
• nod-ai / transformer-benchmarks
  benchmarking some transformer deployments
  ☆26Updated last year
• jax-ml / ml_dtypes
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.
  ☆208Updated this week
• google-deepmind / einshape
  ☆98Updated 4 months ago
• nod-ai / SRT
  Nod.ai 🦈 version of 👻 . You probably want to start at https://github.com/nod-ai/shark for the product and the upstream IREE repository …
  ☆107Updated this week
• NVIDIA / free-threaded-python
  No-GIL Python environment featuring NVIDIA Deep Learning libraries.
  ☆20Updated this week
• lianakoleva / no-libtorch-compile
  ☆17Updated 2 weeks 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
• zasdfgbnm / autonvtx
  Automatically insert nvtx ranges to PyTorch models
  ☆17Updated 3 years ago
• Artur-Galstyan / statedict2pytree
  ☆40Updated 4 months ago
• alexzhang13 / Triton-Puzzles-Solutions
  Personal solutions to the Triton Puzzles
  ☆15Updated 3 months ago
• fishmingyu / GeoT
  GeoT: Tensor Centric Library for Graph Neural Network via Efficient Segment Reduction on GPU
  ☆18Updated 2 weeks ago
• gmarkall / life-of-a-numba-kernel
  Worked example of the process from Python source to CUDA kernel execution with Numba
  ☆36Updated last month
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆98Updated last month
• facebookresearch / adaptive_scheduling
  Experimental scripts for researching data adaptive learning rate scheduling.
  ☆23Updated last year
• alvarobartt / safejax
  Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
  ☆42Updated 5 months ago
• ezyang / tlparse
  TORCH_LOGS parser for PT2
  ☆21Updated 3 weeks ago
• pytorch-labs / torchfix
  TorchFix - a linter for PyTorch-using code with autofix support
  ☆98Updated last month