from torch._subclasses.fake_tensor import FakeTensorMode with FakeTensorMode(): pass Traceback (most recent call last): File "/data/users/weif/torchtitan/test_fake ...
This Python package implements the Oliker-Prussner-Method [1] for solving a two dimensional version of the Monge-Ampère-Equation: $$\det \left(D^2 u(x,y)\right)=f(x,y)$$ This was originally a project ...