Update titan weights to load in bfloat16 #166
Description
This landed - pytorch/torchtitan#1646 - so we can cut our memory usage in half once it is in the nightly builds.
{{ message }}
This landed - pytorch/torchtitan#1646 - so we can cut our memory usage in half once it is in the nightly builds.