nixpkgs/pkgs/development/python-modules/jax
Samuel Ainsworth 2f4ac6f803 python3Packages.jax: add CUDA tests in passthru
Organize CUDA-enabled tests for jax and jaxlib into passthru scripts to facilitate testing.
2024-02-28 02:50:45 +00:00
..
default.nix
test-cuda.nix