jax-amd-gpu-setup

Run JAX on AMD GPU — Windows DirectML + Linux/WSL2 ROCm. The missing setup guide for RX 5000/6000/7000 series. vmap, grad, jit tested on real hardware.

amd

View on GitHub

0 Stars

0 Forks

0 Watchers

Python Language

30 SrcLog Score

Cost to Build

$8.4K

Market Value

$920

How is this calculated?

Growth over time

2 data points · 2026-04-10 → 2026-04-18

Stars Forks Watchers

💬

How do you feel about this project?

Ask AI about jax-amd-gpu-setup

Question copied to clipboard

What is the ChharithOeun/jax-amd-gpu-setup GitHub project? Description: "Run JAX on AMD GPU — Windows DirectML + Linux/WSL2 ROCm. The missing setup guide for RX 5000/6000/7000 series. vmap, grad, jit tested on real hardware.". Written in Python. Explain what it does, its main use cases, key features, and who would benefit from using it.

Question is copied to clipboard — paste it after the AI opens.

How to clone jax-amd-gpu-setup

Clone via HTTPS

git clone https://github.com/ChharithOeun/jax-amd-gpu-setup.git

Clone via SSH

[email protected]:ChharithOeun/jax-amd-gpu-setup.git

Download ZIP

Download master.zip

Found an issue?

Report bugs or request features on the jax-amd-gpu-setup issue tracker:

Open GitHub Issues

Similar to jax-amd-gpu-setup

webpack madge steal wufuc conditioner awesome-vulkan esl zydis staxrip pyopencl coriander lodjs easytimer.js ocl angular-async-loader node-dependency-tree purge-wrangler require-vuejs GatelessGateSharp SimpleSvm set-egpu parenchyma node-precinct TypeScript-AMD-Boilerplate ScanTree awesome-d3d12 Tensile generator-module-boilerplate akase romania-choropleth