How to Apply ML/Neural Network to a function

Question

Good morning!

Working as a Software Engineer in the Disney Team and recently I\'ve got a problem, which I solved but It\'s extremely slow.

Problem

You\'re given a s