Simplify JAX Scalify MNIST examples using jax_scalify.tree methods.…
#363
| Job | Run time |
|---|---|
| 24s | |
| 44s | |
| 1m 8s |
jax_scalify.tree methods.…
#363
| Job | Run time |
|---|---|
| 24s | |
| 44s | |
| 1m 8s |