search

google / maxtext

A simple, performant and scalable Jax LLM!
Apache License 2.0
1.39k stars 247 forks source link

Install Transformer Engine from github for stable and nightly mode #743

Closed michelle-yooh closed 6 days ago

michelle-yooh commented 3 weeks ago

This fixes b/350495728