Integrate new API for saving and loading with neuronx_distributed
(…
#263
Job | Run time |
---|---|
2m 21s | |
2m 21s |
neuronx_distributed
(…
#263
Job | Run time |
---|---|
2m 21s | |
2m 21s |