高级软件工程师 - 机器学习推理和性能

14个月前远程全职
AI Fund

AI Fund

location 旧金山
unsaved
你是否对推动人工智能的前沿感兴趣?我们正在寻找一名高级软件工程师加入我们充满活力的团队。这个角色非常适合那些在快节奏的创业环境中蓬勃发展,并渴望为令人兴奋的机器学习领域做出重大贡献的人。如果你是一名有机器学习经验的后端工程师,现在想把这个作为你的重点发展方向,我们期待收到你的申请。 主要职责: • 扩展、优化和微调我们的机器学习模型及其基础设施的性能 • 深入研究Pytorch、Transformers、CUDA和其他库的底层代码,以调试机器学习性能问题 • 优化和扩展机器学习模型,特别是大型语言模型 • 与多元化团队合作,设计和实施创新解决方案 必备条件: • 3-10年以上一种或多种通用编程语言的经验,包括但不限于Python、Ruby或Go • 对软件工程原理有深入理解,并具有开发和部署人工智能/机器学习推理解决方案的成功经验 • 对机器学习库特别是PyTorch有深入了解 • 具有机器学习和大型语言模型方面的兴趣和经验 • 对GPU架构有深入了解 • 熟悉Docker和Kubernetes 加分项: • 熟练提高软件系统性能的能力,特别是在LLMs的背景下 • 熟悉LLM优化技术(量化、猜测解码、连续批处理等) • 具有CUDA或类似技术的经验非常有优势 我们提供: • 有竞争力的薪酬福利(无限制休假、401k、医疗保险费用报销、远程优先文化) • 一个独特的机会,成为一个迅速发展的创业公司的一部分,参与到我们这个时代最令人兴奋的工程领域中 • 一个包容和支持的工作文化,促进学习和成长 • 接触各种机器学习创业公司,提供无与伦比的学习和人脉机会 立即申请,踏上塑造人工智能未来的有益之旅!如果你是一名有动力、对机器学习充满热情,并渴望成为一个合作和有前瞻性的团队的一部分的人,我们很愿意听到你的声音。 在Baseten,我们致力于营造一个多元包容的工作环境。我们为所有员工和申请人提供平等的就业机会,不论种族、肤色、宗教、性别、性取向、性别认同或表达、国籍、年龄、遗传信息、残疾或退伍军人身份。