5 projects
winjax
Native Windows CUDA support for JAX (unofficial PJRT plugin loader)
winjax-cuda13-plugin
JAX Plugin for NVIDIA GPUs
winjax-cuda13-pjrt
XLA CUDA PJRT plugin DLL for JAX on native Windows (winjax)
metaljax
Experimental Metal backend for JAX on Apple silicon: PJRT plugin executing StableHLO on MLX
vui
GUI framework written in pure Python
