lots of new WIP on the vectorized NN impl