Scalable Mechanistic Neural Networks