Defining new Jax types with hijax4 hours ago#hijax#custom types#JAXhttps://docs.jax.dev/en/latest/hijax_types.htmlCopy Link