Alternatives and similar repositories for jax

Users that are interested in jax are comparing it to the libraries listed below

• ROCm / bitsandbytes
  8-bit CUDA functions for PyTorch
  ☆68Updated last month
• ROCm / ROCmValidationSuite
  A system validation and diagnostics tool for monitoring, stress testing, detecting, and troubleshooting issues impacting AMD GPUs in high…
  ☆88Updated last week
• ROCm / ROCT-Thunk-Interface
  ROCm's Thunk Interface
  ☆91Updated 8 months ago
• ROCm / rocAL
  The AMD rocAL is designed to efficiently decode and process images and videos from a variety of storage formats and modify them through a…
  ☆22Updated this week
• ROCm / rocm-cmake
  CMake modules used within the ROCm libraries
  ☆68Updated this week
• ROCm / ROCR-Runtime
  [DEPRECATED] Moved to ROCm/rocm-systems repo
  ☆269Updated this week
• intel / xpumanager
  ☆139Updated last month
• ROCm / apex
  A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
  ☆23Updated last week
• ROCm / rocm_bandwidth_test
  Bandwidth test for ROCm
  ☆69Updated last week
• ROCm / triton
  Development repository for the Triton language and compiler
  ☆137Updated this week
• ROCm / roctracer
  [DEPRECATED] Moved to ROCm/rocm-systems repo
  ☆84Updated 3 weeks ago
• amd / UIF
  ☆61Updated 2 years ago
• intel / intel-extension-for-openxla
  ☆51Updated last week
• artyom-beilis / dlprimitives
  Deep Learning Primitives and Mini-Framework for OpenCL
  ☆204Updated last year
• ROCm / rocm_smi_lib
  [DEPRECATED] Moved to ROCm/rocm-systems repo
  ☆149Updated 2 weeks ago
• intel / intel-extension-for-deepspeed
  Intel® Extension for DeepSpeed* is an extension to DeepSpeed that brings feature support with SYCL kernels on Intel GPU(XPU) device. Note…
  ☆63Updated 4 months ago
• ROCm / hipBLASLt
  [DEPRECATED] Moved to ROCm/rocm-libraries repo
  ☆114Updated this week
• ROCm / rocBLAS
  [DEPRECATED] Moved to ROCm/rocm-libraries repo
  ☆388Updated this week
• ROCm / rccl-tests
  RCCL Performance Benchmark Tests
  ☆79Updated this week
• oneapi-src / level-zero-tests
  oneAPI Level Zero Conformance & Performance test content
  ☆57Updated 2 weeks ago
• ROCm / vllm
  A high-throughput and memory-efficient inference and serving engine for LLMs
  ☆108Updated this week
• ROCm / xformers
  Hackable and optimized Transformers building blocks, supporting a composable construction.
  ☆32Updated last week
• ROCm / rocm-blogs
  ☆67Updated this week
• ROCm / rocm-examples
  A collection of examples for the ROCm software stack
  ☆253Updated this week
• ROCm / ROCgdb
  This is ROCgdb, the ROCm source-level debugger for Linux, based on GDB, the GNU source-level debugger.
  ☆63Updated this week
• ROCm / AMDMIGraphX
  AMD's graph optimization engine.
  ☆266Updated this week
• ROCm / clr
  ☆153Updated this week
• ROCm / hipBLAS
  [DEPRECATED] Moved to ROCm/rocm-libraries repo
  ☆147Updated this week
• ROCm / TransferBench
  TransferBench is a utility capable of benchmarking simultaneous copies between user-specified devices (CPUs/GPUs)
  ☆51Updated last week
• ROCm / flash-attention
  Fast and memory-efficient exact attention
  ☆200Updated last month