Defining new Jax types with hijax(docs.jax.dev)
12 pointsby fhchl4 hours ago2 comments
  • 0xnyn3 hours ago
    Hijacking JAX with hijax. Such a neat name.
  • chrisjj3 hours ago
    > the usual tool is a pytree:

    404.