io+hdf5+lazy-loading+processing-pipelines for EHRs and generic PyTrees for JAX friendly tasks, mainly ML