14 points | by fhchl 12 hours ago
3 comments
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.
Hijacking JAX with hijax. Such a neat name.
> the usual tool is a pytree:
404.
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.
Hijacking JAX with hijax. Such a neat name.
> the usual tool is a pytree:
404.