alexander-g / vkJAXLinks
JAX interpreter for Vulkan
☆16Updated 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 last month
- ☆91Updated this week
- benchmarking some transformer deployments☆26Updated last month
- npcomp - An aspirational MLIR based numpy compiler☆51Updated 5 years ago
- ☆137Updated last week
- A lightweight memory allocator for hardware-accelerated machine learning☆181Updated 4 months ago
- Open source cross-platform compiler for compute-intensive loops used in AI algorithms, from Microsoft Research☆116Updated 2 years ago
- Notes and artifacts from the ONNX steering committee☆28Updated last week
- Machine Learning Framework for Operating Systems - Brings ML to Linux kernel☆253Updated 4 years ago
- Repo for AI Compiler team. The intended purpose of this repo is for implementation of a PJRT device.☆51Updated this week
- ☆66Updated last year
- IREE's PyTorch Frontend, based on Torch Dynamo.☆105Updated last week
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆26Updated 3 weeks ago
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆148Updated 2 weeks ago
- Awesome utilities for performance profiling☆199Updated 11 months ago
- A tracing JIT compiler for PyTorch☆13Updated 4 years ago
- [DEPRECATED] Moved to ROCm/rocm-systems repo☆84Updated 3 weeks ago
- User-Mode Driver for Tenstorrent hardware☆36Updated this week
- Open deep learning compiler stack for cpu, gpu and specialized accelerators☆35Updated 3 years ago
- A lightweight MLIR Python frontend with support for PyTorch☆29Updated last year
- ☆49Updated 5 years ago
- NNCG: A Neural Network Code Generator☆35Updated last year
- An experimental CPU backend for Triton (https//github.com/openai/triton)☆49Updated 5 months ago
- SynapseAI Core is a reference implementation of the SynapseAI API running on Habana Gaudi☆42Updated last year
- This is ROCgdb, the ROCm source-level debugger for Linux, based on GDB, the GNU source-level debugger.☆66Updated last week
- ☆167Updated this week
- Unified compiler/runtime for interfacing with PyTorch Dynamo.☆104Updated last month
- CMake modules used within the ROCm libraries☆73Updated 2 weeks ago
- Python bindings for UCX☆139Updated 4 months ago
- GPUOcelot: A dynamic compilation framework for PTX☆219Updated last year