大模型训练引擎工程师
岗位描述:负责主流大模型训练框架在壁仞芯片平台上的适配、功能验证、问题排查与性能调优;支撑预训练、微调等核心训练链路建设,落地MoE、多模态、智能驾驶等业务场景;定位并解决精度、显存、通信、算子等训练问题,搭建自动化工程体系;搭建自动化体系,借助AI Agent赋能适配工作与训练生态。 岗位要求:熟悉PyTorch核心训练流程与分布式并行策略(DP、TP、PP、ZeRO、FSDP等);掌握Megatron、DeepSpeed等一种或多种...
第 19 页,共 213 页
岗位描述:负责主流大模型训练框架在壁仞芯片平台上的适配、功能验证、问题排查与性能调优;支撑预训练、微调等核心训练链路建设,落地MoE、多模态、智能驾驶等业务场景;定位并解决精度、显存、通信、算子等训练问题,搭建自动化工程体系;搭建自动化体系,借助AI Agent赋能适配工作与训练生态。 岗位要求:熟悉PyTorch核心训练流程与分布式并行策略(DP、TP、PP、ZeRO、FSDP等);掌握Megatron、DeepSpeed等一种或多种...
岗位描述:基于壁仞GPU软件栈,开发大模型和多模态大模型推理服务框架;围绕并行策略、GPU通信、算子融合、量化压缩等方向,持续优化大语言模型推理性能,降低模型推理延迟,提升线上模型服务吞吐。 岗位要求:硕士及以上学历,计算机、数学、软件工程、自动化、通信、微电子等相关专业,具有3年及以上相关工作经历;具备扎实编程能力,熟练使用C/C++或Python;熟悉大模型推理技术栈,有vLLM、SGLang、LMDeploy、TensorRT-L...
岗位描述:基于壁仞GPU生态,实现大模型推理服务框架,进行特性覆盖和持续升级,包括vLLM、SGLang等主流方向;基于壁仞GPU生态,通过量化、算子融合、通信优化、并行策略、投机采样、PD分离等多种技术手段,持续提升大模型性能,提供高吞吐的线上服务能力。 岗位要求:硕士及以上学历,计算机、数学、软件工程、自动化、通信、微电子等相关专业,具有3年以上相关工作经历;熟悉深度学习算法,有vLLM、SGLang或TensorRT等推理引擎开发...
岗位描述:支持自研 GPU 在 AI 和大模型场景下的编译优化与性能提升;负责面向自研 GPU 的编译优化方案与算法设计和开发;负责制定并实施自研 GPU 计算核部分的基准性能评估与编译优化性能评估方案;参与未来自研 GPU 处理器指令集和微架构设计,演进编译器优化能力以支持新特性。 岗位要求:计算机软件或相关专业硕士及以上学历,精通 C/C++ 编程,熟悉计算机体系结构和编译原理;具备良好的沟通能力和代码习惯,熟悉 Linux 开发调...
岗位描述:设计、开发和调试支持多 GPU 的驱动程序和软件;与其他软件工程师共同完成驱动和支撑软件的设计与开发,提供对多 GPU(南向互联和北向互联)的支持;与软硬件工程师协调解决软硬件架构问题,为下一代芯片提供高性能、低延迟、高可靠性的大集群软件解决方案;协助客户平台的多 GPU 方案落地,并根据用户需求完善整体平台端到端软件方案;与其他软硬件工程师协同,解决多 GPU 平台中的问题并优化系统性能。 岗位要求:具备交换机、路由器、大集...
岗位描述:针对自研架构设计和优化 Kernel 综合器,支持硬件架构验证及高性能算子交付;负责核心基础设施的维护与演进,包括多层 IR 及 Lower 过程、编程范式抽象、面向 Tensor 的抽象、ISA 层抽象;负责关键 Pass 的设计与优化,如生命周期/活跃性分析、资源分配、Peephole 优化;建设与维护回归体系,包括单元测试、集成测试、CI 稳定性、代码规范与静态检查;与硬件团队协同设计,反馈自研指令集在性能与易用性方面的...
岗位描述:负责 AI 编译器的设计与实现;开展 AI 模型性能分析与优化,定位性能瓶颈并改进编译优化算法;解决 AI 模型的编译和部署问题;负责算子适配与调优;承担软件质量维护与功能扩展工作。 岗位要求:具备 XLA、MLIR、TVM、TRITON 等项目中的一个或多个实践经验;熟练使用 C/C++,熟悉常用数据结构和算法,具备良好的编程习惯和代码风格;具备深度学习算法相关知识;计算机、自动化、微电子等相关专业背景;以上要求至少满足两条...
岗位描述:负责将 Kimi 的全球 Discord 社区从单一频道建设为完整生态,让用户愿意留在社区、持续交流并探索产品玩法;制定社区整体策略与运营机制,包括频道架构、分层权限、新手引导和核心用户体系,设计可自运转的社区规则;识别并运营超级用户、开发者、创作者等不同用户圈层,建立版主/Mod 体系;围绕产品发布、模型升级等节点策划 AMA、线上黑客松、Prompt 挑战赛、Show & Tell 等活动;系统化收集用户反馈、Bug 报告...
岗位描述:负责制定 Kimi 的全球整合营销战略,为全球市场建立统一的叙事框架与营销打法,并支持区域团队进行本地化执行;统筹多个市场的 Co-MKT 合作、品牌联合营销与生态伙伴营销,定义可复制的全球增长操作系统;主导与 NVIDIA、AWS、Microsoft 等云厂商、硬件厂商的联合营销,将产品集成、联合发布、开发者活动转化为全球品牌势能;建设全球开发者生态营销,与技术社区、开源生态、开发者大会建立长期合作关系;管理全球品牌一致性,...