Defining new Jax types with hijax

(docs.jax.dev)

10 points | by fhchl 2 hours ago ago

2 comments

0xnyn an hour ago ago

Hijacking JAX with hijax. Such a neat name.
chrisjj an hour ago ago

> the usual tool is a pytree:
404.