Neural network with custom architecture