Got 2d example problem solved in 7 seconds, but actual dataset takes too long