Just to follow up on my own post from yesterday. I posted a bit too early, and had a classic coding mistake present that made the provided code not run. I’ve got it running correctly now, and I do experience around a 15% increase in performance compared to running my model in full float (an a RTX2080Ti card). Thanks all!