Input Convex Neural Network Architecture and Tests

[bunnech]

Adding input convex neural network architecture based on [1] and corresponding tests.

[1] Amos, Brandon, Lei Xu, and J. Zico Kolter. "Input Convex Neural Networks." International Conference on Machine Learning. PMLR, 2017.

[marcocuturi]

Thanks Charlotte! I think you need to sign a CLA so that we can merge.

[marcocuturi]

Thanks a lot Charlotte! before pushing I'd like to make sure we include the flax dependency in the project.

[bunnech]

Thanks Charlotte! I think you need to sign a CLA so that we can merge.

Oh, yes. I signed the CLA.

[marcocuturi]

Thanks Charlotte

[marcocuturi]

Thanks Charlotte! It's great to have this ICNN interface!

[marcocuturi]

can you clean up the utf 8?

[marcocuturi]

Thks!

Issue with internal change on google side.

[marcocuturi]

Thanks!