JAX support for tvm-ffi abi
☆28May 14, 2026Updated 4 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.☆24Aug 24, 2026Updated 3 weeks ago
- System for automated integration of deep learning backends.☆47Aug 15, 2022Updated 4 years ago
- Compact and Agent-Native MoE Training System☆350Updated this week
- Fused KL divergence from hidden states for knowledge distillation☆22Apr 28, 2026Updated 4 months ago
- Open ABI and FFI for Machine Learning Systems☆463Updated this week
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Artifacts for SOSP'19 paper Optimizing Deep Learning Computation with Automatic Generation of Graph Substitutions☆21Apr 15, 2022Updated 4 years ago
- ☆23Aug 26, 2025Updated last year
- Nsight Python is a Python kernel profiling interface based on NVIDIA Nsight Tools☆295Updated this week
- JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training☆83Sep 11, 2026Updated last week
- A home for the final text of all TVM RFCs.☆111Sep 24, 2024Updated last year
- A Python DSL for Apple Metal GPU compute☆26Jul 23, 2026Updated last month
- ☆22May 7, 2025Updated last year
- Tokamax: A GPU and TPU kernel library.☆294Updated this week
- Tilus is a tile-level kernel programming language with explicit control over shared memory and registers.☆492Updated this week
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- LES extension package for NequIP☆21Aug 14, 2026Updated last month
- ☆37Aug 7, 2025Updated last year
- minimal compiler☆24Feb 19, 2026Updated 6 months ago
- ☆51Jul 27, 2026Updated last month
- An extention of TVMScript to write simple and high performance GPU kernels with tensorcore.☆52Jul 23, 2024Updated 2 years ago
- ☆20Dec 15, 2023Updated 2 years ago
- ☆21Mar 3, 2025Updated last year
- ☆145Jan 30, 2025Updated last year
- LoRAFusion: Efficient LoRA Fine-Tuning for LLMs☆30Jul 2, 2026Updated 2 months ago
- Open source password manager - Proton Pass • AdSecurely store, share, and autofill your credentials with Proton Pass, the end-to-end encrypted password manager trusted by millions.
- ☆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 last year
- ☆36Jul 19, 2025Updated last year
- A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer☆96Jun 21, 2026Updated 2 months ago
- DietCode Code Release☆66Jul 21, 2022Updated 4 years ago
- Ship correct and fast LLM kernels to PyTorch☆154Jan 14, 2026Updated 8 months ago
- Benchmark PyTorch Custom Operators☆14Jul 6, 2023Updated 3 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- MSLK (Meta Superintelligence Labs Kernels) is a collection of PyTorch GPU operator libraries that are designed and optimized for GenAI tr…☆154Updated this week
- ☆193Mar 28, 2023Updated 3 years ago
- FLA but cuTile☆27Apr 17, 2026Updated 5 months ago
- An experimental implementation of compiler-driven automatic sharding of models across a given device mesh.☆97Updated this week
- PyTorch/PyG implementation of M3GNet☆12Sep 11, 2023Updated 3 years ago
- Official repository for Parallax (Parameterized Local Linear Attention)☆69Jul 30, 2026Updated last month
- This repository contains code and data for "On the Multimodal Person Verification Using Audio-Visual-Thermal Data"☆12Apr 27, 2023Updated 3 years ago