职位描述
AML是字节跳动公司的机器学习中台,为抖音/今日头条/西瓜视频等业务提供推荐/广告/CV/语音/NLP的训练和推理系统。为公司内业务部门提供强大的机器学习算力,并在这些业务的问题上研究一些具有通用性和创新性的算法。同时,也通过火山引擎将一些机器学习/推荐系统的核心能力提供给外部企业客户。此外,AML还在AI for Science,科学计算等领域做一些前沿研究。
1、负责字节跳动 AML 机器学习在线服务框架的研究与开发,服务于公司各个产品;
2、高效部署,优化计算机视觉、语音识别、语音合成、自然语言处理等字节跳动核心业务模型;
3、与公司各算法部门深度合作,分析业务性能瓶颈和系统架构特征,软硬件结合优化,实现极致性能。
职位要求
1、熟练掌握Linux环境下的C/C++、Python语言;
2、具备扎实的计算机科学功底和编程能力,熟悉常见算法和数据结构,具有良好的编程习惯;
3、能够熟练使用至少一种主流的机器学习框架(TensorFlow / PyTorch等),熟悉框架内部实现;
4、熟悉至少一种经典深度学习模型及其应用场景,如ResNet、BERT等;
5、有良好的工作文档习惯,及时按要求撰写更新工作流程及技术文档;
加分项:
1、对深度学习训练和推理模型调试、调优有实操经验;
2、理解 GPU硬件架构,理解 GPU 软件栈(CUDA,cuDNN),具备 GPU 性能分析的经验;
3、熟悉各类深度学习网络和算子底层实现细节;
4、熟悉并行计算算法,擅长各类并行编程;
5、熟悉模型剪枝、量化等优化方法;
6、熟悉 TVM、MLIR、XLA 等模型优化工具;
7、了解 GAN,强化学习,图神经网络,AutoML等。