Alternatives and similar repositories for jax

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

• ROCm / ROCmValidationSuite
  A system validation and diagnostics tool for monitoring, stress testing, detecting, and troubleshooting issues impacting AMD GPUs in high…
  ☆73Updated this week
• ROCm / bitsandbytes
  8-bit CUDA functions for PyTorch
  ☆53Updated this week
• ROCm / rocm_bandwidth_test
  Bandwidth test for ROCm
  ☆58Updated last month
• ROCm / rocm-cmake
  CMake modules used within the ROCm libraries
  ☆67Updated this week
• intel / intel-extension-for-openxla
  ☆47Updated 3 weeks ago
• ROCm / Megatron-LM
  Ongoing research training transformer models at scale
  ☆23Updated 2 weeks ago
• ROCm / TransferBench
  TransferBench is a utility capable of benchmarking simultaneous copies between user-specified devices (CPUs/GPUs)
  ☆41Updated last week
• ROCm / pytorch-micro-benchmarking
  ☆20Updated 3 months ago
• ROCm / hip-tests
  ☆36Updated this week
• ROCm / amdsmi
  AMD SMI
  ☆71Updated this week
• ROCm / ROCT-Thunk-Interface
  ROCm's Thunk Interface
  ☆91Updated 3 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…
  ☆19Updated this week
• ROCm / triton
  Development repository for the Triton language and compiler
  ☆125Updated this week
• ROCm / rccl
  ROCm Communication Collectives Library (RCCL)
  ☆342Updated this week
• ROCm / rocHPCG
  HPCG benchmark based on ROCm platform
  ☆37Updated 3 weeks ago
• ROCm / apex
  A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
  ☆22Updated 2 weeks ago
• ROCm / rocm-examples
  A collection of examples for the ROCm software stack
  ☆224Updated this week
• ROCm / roctracer
  ROCm Tracer Callback/Activity Library for Performance tracing AMD GPUs
  ☆83Updated last week
• ROCm / rocmProfileData
  ☆25Updated this week
• ROCm / rocm-docs-core
  ROCm Documentation Python package for ReadTheDocs build standardization
  ☆16Updated last week
• ROCm / ROCgdb
  This is ROCgdb, the ROCm source-level debugger for Linux, based on GDB, the GNU source-level debugger.
  ☆59Updated last week
• ROCm / flash-attention
  Fast and memory-efficient exact attention
  ☆174Updated this week
• intel / torch-xpu-ops
  ☆46Updated this week
• ROCm / rccl-tests
  RCCL Performance Benchmark Tests
  ☆68Updated last month
• ROCm / hipBLASLt
  [DEPRECATED] Moved to ROCm/rocm-libraries repo
  ☆106Updated this week
• ROCm / hipSPARSE
  ROCm SPARSE marshalling library
  ☆67Updated this week
• oneapi-src / level-zero-tests
  oneAPI Level Zero Conformance & Performance test content
  ☆54Updated last week
• ROCm / ROCm-Device-Libs
  ROCm Device Libraries
  ☆97Updated last year
• ROCm / TransformerEngine
  ☆38Updated this week
• ROCm / rocWMMA
  rocWMMA
  ☆115Updated last week