The following assert should probably be made into an error instead: https://github.com/coreylowman/dfdx/blob/4722a99d303f347d6088d95867d007c75ca6dd78/dfdx-core/src/tensor/safetensors.rs#L16