NVIDIA BioNeMo Framework v2.6.3
Updates & Improvements
- Fixes numerous issues with Evo2 model:
- ESM2 LoRA model inference issue resolved. #996
- Added experimental evo2-mamba model. #888
- Updated base Docker image to nvidia-pytorch 25.06-py3
- NCCL issue in ESM2 pretraing resolved. #970
What's Changed
- Fix test_train_evo2_stops test by @balvisio in #965
- Enable test_train_evo2_stop_at_max_steps_and_continue. by @balvisio in #966
- automated benchmarks: esm2 650M training analogous to bionemo-recipes by @dorotat-nv in #975
- Fix database path in esm2_pretrain_recipes by @pstjohn in #978
- Add fp8 stop and go test for evo2 by @jwilber in #974
- Update Docs Banner for GitHub Pages-hosted Docs by @tshimko-nv in #981
- Add release notes for v2.6.2 (25.06) by @trvachov in #971
- Evo2 Generation fixes and necessary base dependency and container updates. Large change. by @jwilber in #949
- Point NeMo submodule back to main repo by @trvachov in #984
- Use new b2b kernels in evo2 jet tests by @jwilber in #985
- change where dtype is found in checkpoint export by @pstjohn in #989
- Evo2 Mamba by @jstjohn in #888
- Adding inference CDS length tests by @jstjohn in #991
- Fix PIL CVE by @trvachov in #992
- [BIONEMO-2334] Patch TE to fix Evo2 stop and go training by @balvisio in #987
- Fix bug in evo2-mamba train and add test by @jstjohn in #994
- Fix esm2 lora inference by @yzhang123 in #996
- Reset parameters for the ESM-2 contact head on HF export by @pstjohn in #983
Full Changelog: v2.6.2...v2.6.3