Generate also jax-cuda-plugin and jax-cuda-pjrt in cuda builds and bump CUDA used at built time to 12.6 and add fixes for CUDA #112
| Job | Run time |
|---|---|
| 5h 39m 58s | |
| 5h 39m 58s | |
| 5h 39m 58s | |
| 5h 39m 24s | |
| 5h 39m 24s | |
| 5h 39m 24s | |
| 5h 39m 58s | |
| 5h 39m 58s | |
| 5h 39m 58s | |
| 5h 43m 2s | |
| 5h 43m 2s | |
| 5h 43m 2s | |
| 3h 35m 41s | |
| 3h 35m 41s | |
| 3h 35m 41s | |
| 3h 39m 43s | |
| 3h 39m 43s | |
| 3h 39m 43s | |
| 3h 36m 11s | |
| 3h 36m 11s | |
| 3h 36m 11s | |
| 2h 6m 9s | |
| 1s | |
| 3h 30m 18s | |
| 3h 28m 28s | |
| 3h 28m 28s | |
| 3h 21m 35s | |
| 3h 25m 27s | |
| 3h 25m 27s | |
| 3h 20m 17s | |
| 2h 12m 4s | |
| 2h 12m 4s | |
| 3h 23m 31s | |
| 2h 11m 22s | |
| 2h 11m 22s | |
| 3h 21m 40s | |
| 5d 22h 20m 4s |