HN Explorer

4 comments

[ 2.2 ms ] story [ 25.7 ms ] thread
chrisjj
> the usual tool is a pytree:

404.

0xnyn
Hijacking JAX with hijax. Such a neat name.
peterdsharpe
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.
fhchl
As I understand, hijax types can specify how Jax transformations apply to them, ie.e. how they are differentiated, vmapped, ...