ENH: Rewrite np.load to use Python API