(WIP) Neural network API