关于团队
我们是一个紧密团结的团队，由经验丰富的药物研发者、深度学习研究人员和软件工程师组成，拥有共同的使命——推动生物化学领域的AI创新，发现和开发突破性的疗法，帮助那些患有严重疾病的患者。
Genesis AI团队专注于通过在机器学习、物理学和计算化学的交叉领域进行基础研究，开发用于小分子药物发现的基础模型，并构建强大的软件系统，以支持大规模模拟运行和生成及预测AI模型的训练，这些模型旨在从各种分子数据中学习，利用我们拥有数千个GPU和数万个CPU的集群。
关于职位
我们正在寻找有经验的ML基础设施工程师加入团队，领导工程工作，推动我们在分子系统生成建模方面的ML研究议程，这是我们使命的关键。
作为Genesis的工程师，您将领导我们AI平台和基础设施的快速迭代，解锁之前无法实现的性能、效率和规模的新水平。您将构建大规模分布式训练和推理管道、核心MLOps工具和框架，并优化GPU操作以加速ML模型。
Genesis是一个高度协作和跨职能的环境，您将与我们出色的工程师、研究人员和科学家紧密合作。
您将
- 领导工程工作，专注于AI平台的持续改进，重点是快速构建和迭代可扩展和稳健的分布式基础设施，以进行ML训练、推理和评估。
- 支持跨多个集群和多个云的模型训练和部署，优化吞吐量和成本。
- 优化ML模型和其他工作负载的效率，包括延迟、吞吐量、内存消耗等（例如，通过GPU性能工程），突破当前硬件的可能性。
- 为Genesis基础设施平台的长期愿景做出贡献。
您是
- 不断追求技术卓越的优秀工程师。您能够编写干净的代码，并对您所工作的代码库有深刻理解。
- 在GPU集群上进行大模型的分布式训练和推理方面有丰富经验，并熟悉我们使用的一些核心库和框架：Pytorch、Pytorch Lightning、Pytorch Geometric和Ray。
- 独立思考者，具有强烈的责任感和从基础概念化到最先进实现的工程稳健系统的能力。
- 好奇心强，问题导向型思考者，乐于深入探索AI、物理、化学和生物学交叉的新兴领域，并做出基础性贡献和发现（不需要除ML以外的任何先前经验）。
加分项
- 有使用Kubernetes和Terraform在多个云上构建、维护和调试低级集群基础设施的经验。
- 经验丰富的GPU工程师，能够快速找出性能瓶颈，并为大规模ML工作负载设计高性能代码。
- 具有XLA、Triton、CUDA或类似加速器编程语言和/或深度学习编译器栈的经验。
- 具有以下某些方面的经验：分子系统（蛋白质序列和3D结构、小分子等）、ML力场或其他物理信息模型和方法，或其他应用领域（如3D图形）中的点云数据。
薪酬、福利和津贴
- 具有竞争力的薪酬方案，包括工资和股权。
- 全面的健康福利：医疗、牙科和视力（员工100%覆盖）。
- 401(k)计划。
- 开放（无限制）PTO政策。
- 公司办公室提供免费午餐和晚餐。
- 带薪家庭假（产假和陪产假）。
- 人寿和长期及短期残疾保险。
关于Genesis Molecular AI
Genesis Molecular AI正在开创分子AI基础模型，以开启药物设计和开发的新纪元。公司的生成和预测AI平台GEMS（Genesis分子空间探索）将AI和物理学整合到行业领先的模型中，以生成和优化药物分子，包括用于结构预测的突破性生成扩散模型Pearl。Genesis已从领先的AI、科技和生命科学投资者那里筹集了超过3亿美元，与主要制药合作伙伴签署了多项AI研究合作，并正在部署GEMS以推进内部治疗管道，用于各种高影响目标。
Genesis总部位于加利福尼亚州圣马特奥，在圣地亚哥设有一个完全整合的实验室。我们自豪地成为一个包容性的工作场所和一个平等机会的雇主。
