Nitin Srinivasan c78487d23d Add Github action workflows for running continuous tests with Pytest ...

Changes:
- Adds `wheel_tests.yml` that will be used to run continuous jobs that builds artifacts and runs CPU/CUDA tests. Jobs will run by workflow calls to `build_artifacts.yml`/`pytest_cpu.yml`/`pytest_gpu.yml`.
- Adds testing of CUDA tests on H100 gpus
- Make script executable
- Change the name of GPU scripts and workflows to CUDA to be more clear as to what is being tested
PiperOrigin-RevId: 715500412

2025-01-14 13:10:51 -08:00

..

ISSUE_TEMPLATE

Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax

2024-09-20 07:52:33 -07:00

workflows

Add Github action workflows for running continuous tests with Pytest

2025-01-14 13:10:51 -08:00

dependabot.yml

Update dependabot.yml

2021-08-19 20:21:16 +02:00