Comment by fhchl

As I understand, hijax types can specify how Jax transformations apply to them, ie.e. how they are differentiated, vmapped, ...