| Image | Build | Test |
|---|---|---|
 |
 |
|
 |
 |
 |
 |
 |
 |
 |
 |
|
 |
 |
This repo currently hosts a public CI for JAX on NVIDIA GPUs and covers some JAX libraries like: T5x, PAXML, Transformer Engine, and others to come soon.
| Image | Build | Test |
|---|---|---|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
This repo currently hosts a public CI for JAX on NVIDIA GPUs and covers some JAX libraries like: T5x, PAXML, Transformer Engine, and others to come soon.