Comment by sean-regents

check out Nvidia's version: https://github.com/NVIDIA/NeMo-Fabric