Skip to main content

Ctrl+K

This site covers the old Flax Linen API. [Explore the new Flax NNX API ✨]

Quick start
Flax Basics
Guides
Examples
Glossary
Frequently Asked Questions (FAQ)
Developer notes
The Flax philosophy
How to contribute
API Reference
Flax NNX

.rst

Converting and upgrading

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

previous

Extracting intermediate values

next

Migrating from Haiku to Flax

By The Flax authors

© Copyright 2023, The Flax authors.