Defining new Jax types with hijax - Hacker News

Posted by fhchl 3 hours ago

Defining new Jax types with hijax(docs.jax.dev)

11 points | 2 comments

0xnyn 1 hour ago|

Hijacking JAX with hijax. Such a neat name.

chrisjj 2 hours ago|

> the usual tool is a pytree:

404.