Extend to regression problem

[NimaMojtahedi]

Thank you for this great work! I would like to know that if one can extend your module to regression problem?

[KevinMusgrave]

Thanks 😄

It's been a while since I've worked on this library. I'm sure it's possible to extend it to regression. You'll probably have to dig into the code to really understand it though. I had an ambitious idea of creating lazily-evaluated functions that could be combined together in a flexible way, but this made the code very complicated.