JAX gets custom type definitions via the hijax mechanism

The official JAX documentation introduces hijax, a mechanism that lets developers define custom types that integrate natively with the JAX numerical computing framework. Hijax allows external libraries to create JAX-compatible classes without modifying JAX's core source code. The feature is aimed at advanced users building their own libraries on top of JAX.