We can update our installation instructions now that JAX is easier to install, and also mention that JAX works on Windows pretty well now through WSL2 with GPU support!
We can also add some instructions for people to verify that their JAX installation is correctly using the GPU and that Chromatix has been installed correctly.
GJBoth
commented
3 months ago
We can update our installation instructions now that JAX is easier to install, and also mention that JAX works on Windows pretty well now through WSL2 with GPU support!
We can also add some instructions for people to verify that their JAX installation is correctly using the GPU and that Chromatix has been installed correctly.