Converting and upgrading# Migrating from Haiku to Flax Convert PyTorch models to Flax Migrate checkpointing to Orbax Upgrading my codebase to Optax Upgrading my codebase to Linen RNNCellBase Upgrade Guide Migrate to regular dicts