Niru Maheswaranathan nirum
nirum
/ cell07.ipynb
Created
December 6, 2018 18:56
— forked from szapp/cell07.ipynb
Attempt to recreate figure 3 from Maheswaranathan et al, 2018
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| """Save/load pytrees to disk.""" | |
| import collections | |
| import h5py | |
| import jax | |
| import numpy as np | |
| def save(filepath, tree): | |
| """Saves a pytree to an hdf5 file. |
OlderNewer