Defining new Jax types with hijax

(docs.jax.dev)

11 points | by fhchl 3 hours ago ago

2 comments

  • 0xnyn 2 hours ago

    Hijacking JAX with hijax. Such a neat name.

  • chrisjj 2 hours ago

    > the usual tool is a pytree:

    404.