AI-Hypercomputer / AI-Hypercomputer/maxtext
wandb logging support
Abierto
feature request
- Lenguaje dominante
- Python
- Estrellas
- 2.4k
- Forks
- 607
- Merge medio
- 2 d 19 h
- PR fusionados (30 d)
- 158
Descripción
### Feature or Model Request
HI MaxText developers! Recently I'm using MaxText to train LLMs, and in my fork I noticed that there's a way to patch the metric_logger to enable wandb logging for MaxText.
I wonder if this is a desired feature in MaxText. If yes, I'd love to create a PR for this since I have a working implementation.
### Additional Context
_No response_
Guía de contribución
Evaluación
Este issue todavía no se ha evaluado.