Defining new Jax types with hijax

(docs.jax.dev)

10 points | by fhchl 3 hours ago ago

2 comments

$0xnyn an hour ago

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

> the usual tool is a pytree:
404.