bwasti / better_bindings
Better bindings for Python
☆17Updated 2 years ago
Alternatives and similar repositories for better_bindings:
Users that are interested in better_bindings are comparing it to the libraries listed below
- Loop Nest - Linear algebra compiler and code generator.☆22Updated 2 years ago
- A tracing JIT compiler for PyTorch☆12Updated 3 years ago
- Make triton easier☆44Updated 8 months ago
- Write your code as tree-like expressions, then transform it☆21Updated last year
- Hacks for PyTorch☆18Updated last year
- ☆21Updated 3 months ago
- No-GIL Python environment featuring NVIDIA Deep Learning libraries.☆43Updated last week
- A small library for creating and manipulating custom JAX Pytree classes☆57Updated last year
- PyTorch centric eager mode debugger☆46Updated 2 months ago
- A thin, highly portable toolkit for efficiently compiling dense loop-based computation.☆148Updated 2 years ago
- FlexAttention w/ FlashAttention3 Support☆26Updated 4 months ago
- Standalone commandline CLI tool for compiling Triton kernels☆17Updated 5 months ago
- Automatically insert nvtx ranges to PyTorch models☆17Updated 3 years ago
- ☆100Updated 7 months ago
- benchmarking some transformer deployments☆26Updated last year
- ☆51Updated 6 months ago
- ☆59Updated 2 years ago
- A collection of optimizers, some arcane others well known, for Flax.☆29Updated 3 years ago
- ☆40Updated 2 months ago
- Document parameters using comments☆10Updated 3 years ago
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆43Updated 7 months ago
- Personal solutions to the Triton Puzzles☆17Updated 7 months ago
- A place to store reusable transformer components of my own creation or found on the interwebs☆44Updated this week
- TORCH_LOGS parser for PT2☆32Updated this week
- Extends cloudpickle to dynamically pickle modules from your working directory☆10Updated 5 years ago
- Unifying Python/C++/CUDA memory: Python buffered array ↔️ `std::vector` ↔️ CUDA managed memory☆81Updated last week
- pytest plugin for a better developer experience when working with the PyTorch test suite☆44Updated 3 years ago
- Experiment of using Tangent to autodiff triton☆75Updated last year
- DiCE: The Infinitely Differentiable Monte-Carlo Estimator☆31Updated last year
- ☆18Updated 2 years ago