Defining New JAX Types with Hijax
Utilize Hijax to define new JAX types, ensuring data integrity and specialized operations.
JAX operates primarily with arrays, but sometimes data should be modeled as a distinct type. Hijax types allow for this by subclassing HiType, ensuring internal invariants and enabling specialized operations. For instance, quantized arrays can be defined to maintain their integrity and manage internal relationships effectively.