search

IBM / simulai

A toolkit with data-driven pipelines for physics-informed machine learning.
https://ibm.github.io/simulai/
Apache License 2.0
157 stars 23 forks source link

Extend/lightning adapt #163

Closed Joao-L-S-Almeida closed 11 months ago

Joao-L-S-Almeida commented 11 months ago

First attempt of enforce compatibility between PyTorch Lightning and simulai models. All the explicit references to device placement were hidden were possible.