AI-Hypercomputer/maxtext

Make MaxText as Python Modules

JoeZijunZhou opened this issue · 0 comments

JoeZijunZhou commented

Can we refactor the imports to make MaxText as Python Modules? It's pretty hard for developers to use or develop on top of it.

  • Blocking inference development with JetStream.
  • Hard to integrate with other python libraries or modules.