This is tensorflow demo on implementing FWM/JMI and FWI. The codes are in tf2wrw.ipynb and tf2fwi.ipynb respectively.
The aim is to show the potential of Automatic Differentiation feature for implementing inverse problems (not necessary machine/deep learning) by only defining forward modelling operator and objective function.