-
Benjamin Lefaudeux authored
add a clip gradients util, equivalent to torch's but aware of the sharded states. Add a corresponding unit test
ade312c4
add a clip gradients util, equivalent to torch's but aware of the sharded states. Add a corresponding unit test