11## Model Overview
22
33Training code for this model is in
4- [ ` recipes/esm2_native_te ` ] ( https://github.com/NVIDIA-BioNeMo/bionemo-framework /tree/main/recipes/esm2_native_te )
4+ [ ` recipes/esm2_native_te ` ] ( https://github.com/NVIDIA-BioNeMo/bionemo-recipes /tree/main/recipes/esm2_native_te )
55and
6- [ ` recipes/esm2_accelerate_te ` ] ( https://github.com/NVIDIA-BioNeMo/bionemo-framework /tree/main/recipes/esm2_accelerate_te ) ;
6+ [ ` recipes/esm2_accelerate_te ` ] ( https://github.com/NVIDIA-BioNeMo/bionemo-recipes /tree/main/recipes/esm2_accelerate_te ) ;
77use an AMPLIFY model tag such as ` nvidia/AMPLIFY_120M ` . Use
8- [ ` models/amplify ` ] ( https://github.com/NVIDIA-BioNeMo/bionemo-framework /tree/main/models/amplify )
8+ [ ` models/amplify ` ] ( https://github.com/NVIDIA-BioNeMo/bionemo-recipes /tree/main/models/amplify )
99for checkpoint conversion and model implementation details.
1010
1111### Description
@@ -72,7 +72,7 @@ Schulz, Sarath Chandar, Christopher James Langmead bioRxiv 2024.09.23.614603; do
7272
7373The model is fully compatible with weights distributed via HuggingFace, i.e.,
7474[ chandar-lab/AMPLIFY_120M] ( https://huggingface.co/chandar-lab/AMPLIFY_120M ) . See
75- [ ` models/amplify ` ] ( https://github.com/NVIDIA-BioNeMo/bionemo-framework /tree/main/models/amplify )
75+ [ ` models/amplify ` ] ( https://github.com/NVIDIA-BioNeMo/bionemo-recipes /tree/main/models/amplify )
7676for checkpoint conversion utilities and usage examples.
7777
7878## Training & Evaluation
@@ -98,9 +98,9 @@ AMPLIFY is provided under the Apache 2.0 license.
9898
9999Use the recipe READMEs for current training entrypoints:
100100
101- - [ ` models/amplify ` ] ( https://github.com/NVIDIA-BioNeMo/bionemo-framework /tree/main/models/amplify )
102- - [ ` recipes/esm2_native_te ` ] ( https://github.com/NVIDIA-BioNeMo/bionemo-framework /tree/main/recipes/esm2_native_te )
103- - [ ` recipes/esm2_accelerate_te ` ] ( https://github.com/NVIDIA-BioNeMo/bionemo-framework /tree/main/recipes/esm2_accelerate_te )
101+ - [ ` models/amplify ` ] ( https://github.com/NVIDIA-BioNeMo/bionemo-recipes /tree/main/models/amplify )
102+ - [ ` recipes/esm2_native_te ` ] ( https://github.com/NVIDIA-BioNeMo/bionemo-recipes /tree/main/recipes/esm2_native_te )
103+ - [ ` recipes/esm2_accelerate_te ` ] ( https://github.com/NVIDIA-BioNeMo/bionemo-recipes /tree/main/recipes/esm2_accelerate_te )
104104
105105| Model Size | GPUs | Batch Size (per GPU) | Training Step Time (s) |
106106| ---------- | ---------------- | -------------------- | ---------------------- |
0 commit comments