| ▲ | Defining new Jax types with hijax(docs.jax.dev) | |
| 12 points by fhchl 3 hours ago | 2 comments | ||
| ▲ | 0xnyn 2 hours ago | parent | next [-] | |
Hijacking JAX with hijax. Such a neat name. | ||
| ▲ | chrisjj 2 hours ago | parent | prev [-] | |
> the usual tool is a pytree: 404. | ||