to_jax() and to_numpy() support to receive a unit argument
#118
| Job | Run time |
|---|---|
| 1m 1s | |
| 57s | |
| 1m 58s | |
| 55s | |
| 1m 6s | |
| 1m 58s | |
| 53s | |
| 2m 17s | |
| 59s | |
| 48s | |
| 1m 48s | |
| 47s | |
| 15m 27s |