Defining new Jax types with hijax

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.