Techniques for training large neural networks