Alternatives and similar repositories for jax:

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

• ROCm / Megatron-LM
  Ongoing research training transformer models at scale
  ☆18Updated this week
• ROCm / bitsandbytes
  8-bit CUDA functions for PyTorch
  ☆50Updated this week
• amd / UIF
  ☆60Updated last year
• ROCm / TransferBench
  TransferBench is a utility capable of benchmarking simultaneous copies between user-specified devices (CPUs/GPUs)
  ☆39Updated 2 weeks ago
• ROCm / triton
  Development repository for the Triton language and compiler
  ☆118Updated this week
• ROCm / amdsmi
  AMD SMI
  ☆61Updated this week
• ROCm / apex
  A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
  ☆22Updated last week
• ROCm / xformers
  Hackable and optimized Transformers building blocks, supporting a composable construction.
  ☆30Updated this week
• ROCm / ROCmValidationSuite
  A system validation and diagnostics tool for monitoring, stress testing, detecting, and troubleshooting issues impacting AMD GPUs in high…
  ☆68Updated this week
• intel / intel-extension-for-openxla
  ☆46Updated last week
• intel / xpumanager
  ☆104Updated last week
• ROCm / rocm-cmake
  CMake modules used within the ROCm libraries
  ☆66Updated last week
• 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…
  ☆17Updated this week
• ROCm / rocm_bandwidth_test
  Bandwidth test for ROCm
  ☆54Updated 2 weeks ago
• ROCm / flash-attention
  Fast and memory-efficient exact attention
  ☆171Updated this week
• ROCm / ROCT-Thunk-Interface
  ROCm's Thunk Interface
  ☆88Updated last month
• ROCm / hipBLASLt
  hipBLASLt is a library that provides general matrix-matrix operations with a flexible API and extends functionalities beyond a traditiona…
  ☆91Updated this week
• ROCm / rocHPL
  High Performance Linpack for Next-Generation AMD HPC Accelerators
  ☆52Updated 2 weeks ago
• ROCm / rccl-tests
  RCCL Performance Benchmark Tests
  ☆64Updated this week
• ROCm / rocWMMA
  rocWMMA
  ☆109Updated this week
• ROCm / MAD
  ☆18Updated this week
• ROCm / rdc
  RDC
  ☆28Updated this week
• ROCm / rocSHMEM
  rocSHMEM intra-kernel networking runtime for AMD dGPUs on the ROCm platform.
  ☆77Updated this week
• ROCm / AITemplate
  AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (N…
  ☆11Updated 10 months ago
• intel / intel-xpu-backend-for-triton
  OpenAI Triton backend for Intel® GPUs
  ☆182Updated this week
• ROCm / roctracer
  ROCm Tracer Callback/Activity Library for Performance tracing AMD GPUs
  ☆83Updated this week
• intel / torch-xpu-ops
  ☆38Updated this week
• ROCm / rocHPCG
  HPCG benchmark based on ROCm platform
  ☆37Updated last month
• ROCm / vllm
  A high-throughput and memory-efficient inference and serving engine for LLMs
  ☆75Updated this week
• ROCm / hipBLAS
  ROCm BLAS marshalling library
  ☆138Updated this week