maxtext
maxtext copied to clipboard
[WIP] add batch inference