部门是字节跳动火山引擎的机器学习平台
基础要求:
- 熟悉操作系统
- 熟悉 pytorch
- 了解 cuda 编程
- 性能开朗
加分点:
- 对于 pytorch profiler 、nsight system/compute 之类的工具有过使用经验
- 对于训练推理框架开源社区有过了解,阅读过 megatron lm 、vllm 、xdit 之类热门项目的代码
工作内容:
- 主要工作是对训练做调优,模型、infra 的调优都会涉及,包括一些底层的 profiling 能力和算子优化
联系方式: [email protected]