The latest stable release of
GPJax can be installed via
Check your installation
We recommend you check your installation version:
Fancy using GPJax on GPU/TPU? Then you'll need to install JAX with the relevant hardware acceleration support as detailed in the JAX installation guide.
This version is possibly unstable and may contain bugs.
The latest development version of
GPJax can be installed via running following: