v1.2.0

github-actions released this 08 Mar 06:25

ef4ea98

Fix kd.sharding.FSDPSharding() to supports jax.ShapeDtypeStruct
kd.data:
- kd.data.py.PyGrainPipeline supports direct indexing (ds[0]).
- kd.data.py.HuggingFace supports
Typeguard / typechecking
+various changes and improvements

Assets 2