orbax.experimental.model API Reference

• model.jax2obm.main_lib
• model.core.python.concrete_function
• model.core.python.constant
• model.core.python.file_utils
• model.core.python.function
• model.core.python.manifest_util
• model.core.python.polymorphic_function
• model.core.python.shlo_function
• model.core.python.signature
• model.core.python.tree_util
• model.core.python.type_proto
• model.core.python.unstructured_data
• model.core.python.value
• model.tf2obm.tf_function_def_to_obm