jax

8514d7b4 - Allow deserialization with a dtype and Format (layout) via a ShapeDtypeStruct spec