bionemo-core
lightning>=2.2.1
megatron-core
nemo_toolkit[all]
pytorch-lightning>=2.2.1

[post]
transformer_engine[pytorch]

[test]
bionemo-testing
