Hacker News new | ask | show | jobs
Defining new Jax types with hijax (docs.jax.dev)
14 points by fhchl 20 days ago
3 comments
peterdsharpe 20 days ago
Why use this over Equinox's Module, plus jaxtyping? That seems like the cleaner (and more established) solution, but maybe there are pros/cons I'm not seeing.
link
fhchl 19 days ago
As I understand, hijax types can specify how Jax transformations apply to them, ie.e. how they are differentiated, vmapped, ...
link
0xnyn 20 days ago
Hijacking JAX with hijax. Such a neat name.
link
chrisjj 20 days ago
> the usual tool is a pytree:

404.

link