bionemo-core
lightning>=2.2.1
megatron-core
nemo_toolkit[multimodal,nlp]
pyzmq==26.4.0
pytorch-lightning>=2.2.1

[post]
transformer_engine[pytorch]

[test]
bionemo-testing
