alexander-g / vkJAXLinks
JAX interpreter for Vulkan
โ15Updated 4 years ago
Alternatives and similar repositories for vkJAX
Users that are interested in vkJAX are comparing it to the libraries listed below
Sorting:
- 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 3 weeks ago
- โ130Updated this week
- GPUOcelot: A dynamic compilation framework for PTXโ218Updated 10 months ago
- benchmarking some transformer deploymentsโ26Updated 3 weeks ago
- Bandwidth test for ROCmโ71Updated this week
- Open deep learning compiler stack for cpu, gpu and specialized acceleratorsโ35Updated 3 years ago
- โ85Updated this week
- A lightweight MLIR Python frontend with support for PyTorchโ29Updated last year
- โ318Updated 3 weeks ago
- [DEPRECATED] Moved to ROCm/rocm-systems repoโ84Updated this week
- Notes and artifacts from the ONNX steering committeeโ27Updated this week
- A tracing JIT compiler for PyTorchโ13Updated 4 years ago
- Open source cross-platform compiler for compute-intensive loops used in AI algorithms, from Microsoft Researchโ113Updated 2 years ago
- SynapseAI Core is a reference implementation of the SynapseAI API running on Habana Gaudiโ42Updated 10 months ago
- A profiler to disclose and quantify hardware features on GPUs.โ175Updated 3 years ago
- User-Mode Driver for Tenstorrent hardwareโ36Updated this week
- Issues related to MLPerfยฎ Inference policies, including rules and suggested changesโ64Updated last month
- CMake modules used within the ROCm librariesโ70Updated this week
- Unified compiler/runtime for interfacing with PyTorch Dynamo.โ104Updated 2 weeks ago
- AMD ROCm Performance Primitives (RPP) library is a comprehensive high-performance computer vision library for AMD processors with HIP/Opeโฆโ68Updated this week
- AMD's graph optimization engine.โ267Updated this week
- A GPU-driven system framework for scalable AI applicationsโ122Updated 10 months ago
- [DEPRECATED] Moved to ROCm/rocm-libraries repoโ114Updated this week
- An experimental CPU backend for Triton (https//github.com/openai/triton)โ47Updated 3 months ago
- npcomp - An aspirational MLIR based numpy compilerโ51Updated 5 years ago
- Awesome utilities for performance profilingโ198Updated 9 months ago
- TVM stack: exploring the incredible explosion of deep-learning frameworks and how to bring them togetherโ64Updated 7 years ago
- IREE's PyTorch Frontend, based on Torch Dynamo.โ102Updated last week
- Repo for AI Compiler team. The intended purpose of this repo is for implementation of a PJRT device.โ45Updated this week
- A lightweight memory allocator for hardware-accelerated machine learningโ176Updated 2 months ago