Closed DhairyaLGandhi closed 3 years ago
Okay this seems to be working. Thanks @maleadt for all the help!
This PR has gotten pretty big, so the DiffEqFlux tests can be done in the next one. It seems to fail with FFJORD not being GPU compatible. I get the Cannot take CPU address...
error.
Disables most tests temporarily so that its easier to see results