search

aws-neuron / neuronx-distributed

MIT No Attribution
30 stars 5 forks source link

Import rank_zero_warn #18

Open joelamzn opened 3 months ago

joelamzn commented 3 months ago

Nit: Import missing rank_zero_warn from lightning/module.py