FlexModel
latest

Getting Started

  • Introduction
  • Examples
  • Demos

Key Python API:

  • Core: Wrapper and HookFunction
  • Distributed: Backend, mappings and strategies
  • Traverse: Pytree traversal utility
FlexModel
  • Demos
  • Edit on GitHub

Demos

  1. Induction Heads Demo

Previous Next

© Copyright 2023, Matthew Choi. Revision 94e3cb43.

Built with Sphinx using a theme provided by Read the Docs.