alexander-g / vkJAX

Related projects ⓘ

Alternatives and complementary repositories for vkJAX

• spectral-compute / scale-examples
  ☆48Updated 4 months ago
• AMYPAD / CuVec
  Unifying Python/C++/CUDA memory: Python buffered array ↔️ `std::vector` ↔️ CUDA managed memory
  ☆81Updated 2 weeks ago
• ROCm / amdsmi
  AMD SMI
  ☆41Updated this week
• ROCm / hipRAND
  Random number library that generate pseudo-random and quasi-random numbers.
  ☆24Updated this week
• ROCm / rocm-cmake
  CMake modules used within the ROCm libraries
  ☆58Updated last week
• nod-ai / SRT
  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 this week
• oneapi-src / SYCLomatic
  ☆228Updated this week
• amd / ZenDNN
  ☆83Updated 5 months ago
• CHIP-SPV / chipStar
  chipStar is a tool for compiling and running HIP/CUDA on SPIR-V via OpenCL or Level Zero APIs.
  ☆226Updated this week
• ROCm / hipBLASLt
  hipBLASLt is a library that provides general matrix-matrix operations with a flexible API and extends functionalities beyond a traditiona…
  ☆61Updated this week
• nod-ai / PI
  A lightweight MLIR Python frontend with support for PyTorch
  ☆20Updated 2 months ago
• ROCm / rocMLIR
  ☆128Updated this week
• iree-org / iree-turbine
  IREE's PyTorch Frontend, based on Torch Dynamo.
  ☆53Updated this week
• ROCm / roctracer
  ROCm Tracer Callback/Activity Library for Performance tracing AMD GPUs
  ☆74Updated last week
• Pylir / Pylir
  An optimizing ahead-of-time Python Compiler
  ☆158Updated 5 months ago
• octoml / Apple-M1-BERT
  3X speedup over Apple’s TensorFlow plugin by using Apache TVM on M1
  ☆135Updated 2 years ago
• ROCm / rocm_bandwidth_test
  Bandwidth test for ROCm
  ☆47Updated last week
• ROCm / hipBLAS
  ROCm BLAS marshalling library
  ☆118Updated this week
• microsoft / Accera
  Open source cross-platform compiler for compute-intensive loops used in AI algorithms, from Microsoft Research
  ☆101Updated last year
• openxla / shardy
  MLIR-based partitioning system
  ☆36Updated this week
• data-apis / array-api-comparison
  Data and tooling to compare the API surfaces of various array libraries.
  ☆54Updated 5 months ago
• ROCm / AMDMIGraphX
  AMD's graph optimization engine.
  ☆185Updated this week
• microsoft / onnxruntime-tvm
  Open deep learning compiler stack for cpu, gpu and specialized accelerators
  ☆33Updated last year
• nod-ai / SHARK-ModelDev
  Unified compiler/runtime for interfacing with PyTorch Dynamo.
  ☆95Updated this week
• ROCm / triton
  Development repository for the Triton language and compiler
  ☆93Updated this week
• intel / vc-intrinsics
  ☆54Updated 2 weeks ago
• ROCm / rocRAND
  RAND library for HIP programming language
  ☆110Updated this week
• spcl / pymlir
  Python interface for MLIR - the Multi-Level Intermediate Representation
  ☆224Updated 3 weeks ago
• NVIDIA / atex
  A TensorFlow Extension: GPU performance tools for TensorFlow.
  ☆25Updated last year