The biggest challenge posed by AI training is in moving the massive datasets between the memory and processor.