Skip to content

0.7.0

Compare
Choose a tag to compare
@cgarciae cgarciae released this 22 Feb 20:03
· 51 commits to master since this release

Features

  • init now only called once internally and required to be called explicitly by the user under certain circumstances
  • summary now uses jax.eval_shape under the hood so its super fast since it doesn't allocate memory or perform any computations on the device.

Merged pull requests:

  • Fix notebook #166 (cgarciae)
  • Single Initialization: Removes the current progressive initialization in favor of a single underlying call to init_step. #165 (cgarciae)