Following is my new blog post. This time I played a bit with the new beta version of TF and implemented a simple model where y is the sum of K gaussians which parameters are learned.

http://zachmoshe.com/2019/08/01/sum-of-gaussians-layer-with-keras-2.0.html