Defining new Jax types with hijax

 (docs.jax.dev)

10 points | by fhchl 2 hours ago

2 comments

  • 0xnyn 59 minutes ago
    Hijacking JAX with hijax. Such a neat name.
  • chrisjj 1 hour ago
    > the usual tool is a pytree:

    404.