NVIDIA/TorchFort v0.3.1

v0.3.1

Repository: NVIDIA/TorchFort

Tag: v0.3.1

Published: 2025-09-15T20:14:55Z

Prerelease: no

Release notes:

What's Changed

This version of TorchFort contains a few minor updates and bug fixes. This version addresses some deficiencies in the existing Fortran interfaces, adds 64-bit integer handling to the NCCL/MPI interfaces used for distributed training and fixes a bug where the extra loss argument tensor for custom loss functions was not automatically moved to the model device. Other minor improvements are removing the dependency on HDF5 in the Fortran examples to simplify builds and adding a missing compiler define causing the cart pole example to unconditionally run on CPU.

Breaking changes

None.

Deprecations

None.

Notable PRs included in this release

• Fortran interface fixes (https://github.com/NVIDIA/TorchFort/pull/56)
• adding kLong support to TorchFort MPI and NCCL wrappers (https://github.com/NVIDIA/TorchFort/pull/58, https://github.com/NVIDIA/TorchFort/pull/59)
• Add missing ENABLE_GPU define to cart pole example build. (https://github.com/NVIDIA/TorchFort/pull/62)
• Add missing calls to move loss module and extra_loss_args to model device. (https://github.com/NVIDIA/TorchFort/pull/63)
• Removing HDF5 dependency from examples to simplify builds. (https://github.com/NVIDIA/TorchFort/pull/66)

Full Changelog: https://github.com/NVIDIA/TorchFort/compare/v0.3.0...v0.3.1