Belle 2 NN inference