Need help with engine shutoffs after driving quickly

Hey folks. I have a 2013 335i xDrive (N55 PWG) with an intake, intercooler, charge pipe, and high flow catted downpipe running BM3 stage 2.

I've had a strange issue where if I accelerate hard, the engine will shut off after. After the shutoff, I have to wait a while before I can re-start the engine.

During this shutdown, i get a drivetrain malfunction and engine code but have been unable to catch it as the scanners I have require the car to start from fully off which clears the code.

I have changed the battery and coded in the new one and had a shop change my eccentric shaft sensor/motor. It seemed to help but now the issue is back. Has anyone seen this sort of problem before?