Optimized inference/training