Skip to content

maxtext-v0.2.1

Latest

Choose a tag to compare

@SurbhiJainUSC SurbhiJainUSC released this 23 Mar 22:05
· 303 commits to main since this release
  • Use the new maxtext[runner] installation option to build Docker images without cloning the repository. This can be used for scheduling jobs through XPK. See the MaxText installation instructions for more info.
  • Config can now be inferred for most MaxText commands. If you choose not to provide a config, MaxText will now select an appropriate one.
  • Configs in MaxText PyPI will now be picked up without storing them locally.
  • New features from DeepSeek-AI are now supported: Conditional Memory via Scalable Lookup (Engram) and Manifold-Constrained Hyper-Connections (mHC). Try them out with our deepseek-custom starter config.
  • MaxText now supports customizing your own mesh and logical rules. Two examples guiding how to use your own mesh and rules for sharding are provided in the custom_mesh_and_rule directory.