Integrate new API for saving and loading with neuronx_distributed
#1243
Job | Run time |
---|---|
32m 43s | |
32m 43s |
neuronx_distributed
#1243
Job | Run time |
---|---|
32m 43s | |
32m 43s |