search

tum-pbs / pbdl-book

Welcome to the Physics-based Deep Learning Book (v0.2)
https://physicsbaseddeeplearning.org/
993 stars 155 forks source link

Is a PyTorch version in the horizon? #4

Closed ritog closed 3 years ago

ritog commented 3 years ago

This project piques my interest, and I would like to read through the book. I have worked on projects with TF/Keras before.

But I would really like a PyTorch version.

Do you have any updates on that?

I would be able to suggest this book to many people if it had a PyTorch version.

Thanks for this great project.

thunil commented 3 years ago

Actually, some of the notebooks are already using pytorch, e.g.: https://www.physicsbaseddeeplearning.org/diffphys-code-ns.html

That being said, at the moment we're not planning to provide every notebook for both TF and pytorch. phiflow (https://github.com/tum-pbs/PhiFlow), which is used for all the differentiable physics examples, supports pytorch, TF and JAX, though. So it shouldn't be too hard to get the code examples working in pytorch.

ritog commented 3 years ago

@thunil thanks. Appreciate the info.