JAX support for tvm-ffi abi
☆26May 14, 2026Updated 2 months ago
Alternatives and similar repositories for jax-tvm-ffi
Users that are interested in jax-tvm-ffi are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Microbenchmarking hyperparameter tuning for JAX functions.☆24Jul 7, 2026Updated last month
- a size profiler for cuda binary☆71Jan 15, 2026Updated 6 months ago
- System for automated integration of deep learning backends.☆47Aug 15, 2022Updated 3 years ago
- Compact and Agent-Native MoE Training System☆327Jul 31, 2026Updated last week
- Fused KL divergence from hidden states for knowledge distillation☆21Apr 28, 2026Updated 3 months ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- Open ABI and FFI for Machine Learning Systems☆445Updated this week
- Artifacts for SOSP'19 paper Optimizing Deep Learning Computation with Automatic Generation of Graph Substitutions☆21Apr 15, 2022Updated 4 years ago
- ☆21Aug 26, 2025Updated 11 months ago
- Nsight Python is a Python kernel profiling interface based on NVIDIA Nsight Tools☆285Updated this week
- JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training☆83Updated this week
- A Python DSL for Apple Metal GPU compute☆25Jul 23, 2026Updated 2 weeks ago
- A home for the final text of all TVM RFCs.☆111Sep 24, 2024Updated last year
- Tokamax: A GPU and TPU kernel library.☆259Updated this week
- Tilus is a tile-level kernel programming language with explicit control over shared memory and registers.☆490Jul 5, 2026Updated last month
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- LES extension package for NequIP☆18Updated this week
- High-performance GPU kernels written in TIRx.☆82Updated this week
- ☆40Dec 14, 2025Updated 7 months ago
- ☆37Aug 7, 2025Updated last year
- ☆11Apr 5, 2021Updated 5 years ago
- minimal compiler☆24Feb 19, 2026Updated 5 months ago
- ☆47Jul 27, 2026Updated 2 weeks ago
- Autonomous CUDA kernel optimization agent with structured task specs and per-config scoring☆17Jun 17, 2026Updated last month
- An extention of TVMScript to write simple and high performance GPU kernels with tensorcore.☆52Jul 23, 2024Updated 2 years ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- ☆20Dec 15, 2023Updated 2 years ago
- ☆21Mar 3, 2025Updated last year
- ☆143Jan 30, 2025Updated last year
- ☆24Mar 22, 2018Updated 8 years ago
- ACM Class 2017 Computer Architecture☆10Jan 11, 2018Updated 8 years ago
- ☆24Feb 20, 2024Updated 2 years ago
- Automatic differentiation for Triton Kernels☆29Aug 12, 2025Updated 11 months ago
- ☆36Jul 19, 2025Updated last year
- A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer☆96Jun 21, 2026Updated last month
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- DietCode Code Release☆65Jul 21, 2022Updated 4 years ago
- Ship correct and fast LLM kernels to PyTorch☆153Jan 14, 2026Updated 6 months ago
- Benchmark PyTorch Custom Operators☆14Jul 6, 2023Updated 3 years ago
- MSLK (Meta Superintelligence Labs Kernels) is a collection of PyTorch GPU operator libraries that are designed and optimized for GenAI tr…☆145Updated this week
- ☆193Mar 28, 2023Updated 3 years ago
- FLA but cuTile☆27Apr 17, 2026Updated 3 months ago
- An experimental implementation of compiler-driven automatic sharding of models across a given device mesh.☆93Updated this week