BioNeMo Recipes (NVIDIA)

TransformerEngine-accelerated checkpoints and training recipes for scaling biological foundation models (ESM-2, AMPLIFY, Geneformer, CodonFM) from single-GPU prototyping to multi-node FSDP training with FP8/MXFP8/NVFP4 precision, compatible with PyTorch, HF Accelerate, and PyTorch Lightning, plus s…