Google近日发布了名为ML Compiler的AI编程辅助工具,旨在帮助开发者更高效地编写和优化机器学习代码。该工具基于深度神经网络,能够自动识别代码中的性能瓶颈,并生成针对特定硬件(如TPU、GPU)的优化版本。
ML Compiler支持TensorFlow和PyTorch等主流框架,可无缝集成到现有开发流程中。Google表示,在内部测试中,该工具将模型训练速度提升了30%以上,同时降低了内存占用。目前,ML Compiler已开放有限预览,开发者可通过Google Cloud平台申请试用。