Lazy evaluation framework for use with Flax and Jax with particular focus on allowing for easy definition of learning (and metalearning) procedures.
See g3doc/index.md for more detailed documentation starting point.
This is not an officially supported Google product.