Custom AI accelerators for large neural networks