v0.2.1

Fixes:

• Pin versions of dm-sonnet to fix broken installation.
• Pin version of easydict because of bug in dependency.
• lumi checkpoint import now works in Python < 3.6.
• Added checkpoints_max_keep option to base_config.yml to keep more than 1 checkpoint during training.