accelerate

8644e23b - Refactor and improve model estimator tool (#2581)

Refactor and improve model estimator tool (#2581) * Start * Stash * Mark * Better mixed precision * Can confirm transformerengine * Finish refactor * Update training usage * Slight tweak * Fin * Fixup test * Add comment about FP8