Open KarlisFre opened 1 year ago
This should work with Jax and TensorFlow.
JIT-compilation in PyTorch has many problems and pitfalls and is not actually that much faster. For fast code, I recommend jit-compiling with Jax or TensorFlow.
I was considering integrating phiflow with nimblephysics for solid motion and collision support https://github.com/keenon/nimblephysics. But nimblephysics supports only pytorch.
I'm looking into fixing the PyTorch jit but this seems like a tricky problem. I'll get back to you.
Hi, I am trying to do gradient based training, unfortunately with jit_compile I get an error:
RuntimeError: Cannot insert a Tensor that requires grad as a constant. Consider making it a parameter or input, or detaching the gradient