search

tum-pbs / PhiFlow

A differentiable PDE solving framework for machine learning
MIT License
1.43k stars 193 forks source link

INNs #79

Closed kbali1297 closed 1 year ago

kbali1297 commented 2 years ago

Changed branch name and Pull Request Title to accomodate implementations for jax and tensorflow.

Implemented INNs for pytorch, tensorflow and jax.stax. User can concatenate multiple coupling layers to compose an INN. User can set backward = True and compute input feature map with the existing network from the output feature map. Drawback : input and output feature map shapes for each coupling layer should be identical (bijectivity is important for inversion).

Python naming conventions were satisfied according to comment made previously.

kbali1297 commented 1 year ago

Tests have been added to commit 9d999db.

holl- commented 1 year ago

I'm merging this into 2.3-develop and do a style pass afterwards.