NVIDIA/bionemo-framework
BioNeMo Framework: For building and adapting AI models in drug discovery at scale
Provides pre-optimized training recipes for biological models (ESM2, CodonFM, Llama3) leveraging NVIDIA TransformerEngine for low-precision training (FP8/MXFP8) and distributed strategies like Megatron-FSDP and sequence packing. Integrates with PyTorch, Hugging Face, and NVIDIA's distributed training stack to enable efficient multi-GPU scaling with benchmarked throughput gains (e.g., 2,367 TFLOPS/GPU on ESM2 15B).
679 stars. Actively maintained with 35 commits in the last 30 days.
Stars
679
Forks
126
Language
Jupyter Notebook
License
—
Category
Last pushed
Mar 13, 2026
Commits (30d)
35
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/NVIDIA/bionemo-framework"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Related frameworks
ersilia-os/ersilia
The Ersilia Model Hub, a repository of AI/ML models for infectious and neglected disease research.
PNNL-CompBio/coderdata
Dataset package for facile training and testing of machine learning/AI algorithms that predict...
adosar/aidsorb
Python package for deep learning on molecular point clouds.
RyanWangZf/PyTrial
PyTrial: A Comprehensive Platform for Artificial Intelligence for Drug Development
CDDLeiden/DrugEx
De Novo Drug Design with RNNs and Transformers