Commit History

feat(train): no batch dimension with pjit
df1fe19

boris commited on

Tokenizer, config, model can be loaded from wandb.
7e48337

Pedro Cuenca commited on

feat: support pypi
f5dba1e

boris commited on