search

NVIDIA / TorchFort

An Online Deep Learning Interface for HPC programs on NVIDIA GPUs
https://nvidia.github.io/TorchFort/
Other
154 stars 19 forks source link

Add train and inference functions for 5d Fortran arrays. #17

Closed nloppi closed 2 months ago

nloppi commented 2 months ago

For 3D grids and convolutions.

azrael417 commented 2 months ago

Looks good to me, thank you very much for adding this, we very much appreciate it.