helmholtz-analytics / heat

Distributed tensors and Machine Learning framework with GPU and MPI acceleration in Python
https://heat.readthedocs.io/
MIT License
211 stars 54 forks source link

Test implementation on JUWELS #1712

Closed ClaudiaComito closed 2 weeks ago

ClaudiaComito commented 2 weeks ago

Code here:

https://gitlab.jsc.fz-juelich.de/haf/dataloader_bachelor_project/-/tree/benchmarking?ref_type=heads