[–] peterdsharpe 1d 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. [–] fhchl 1d ago ↗ As I understand, hijax types can specify how Jax transformations apply to them, ie.e. how they are differentiated, vmapped, ...
[–] fhchl 1d ago ↗ As I understand, hijax types can specify how Jax transformations apply to them, ie.e. how they are differentiated, vmapped, ...
4 comments
[ 1.6 ms ] story [ 30.6 ms ] thread404.