软件工程师,机器学习

旧金山 2天前全职 网络
76.3万 - 129.9万 / 年
NVIDIA 在计算机图形学、PC 游戏和加速计算领域已经转型超过 25 年。这是一个独特的创新遗产,由伟大的技术和出色的人才推动。如今,我们正在利用人工智能的无限潜力来定义计算的下一个时代。在这个时代,我们的 GPU 将作为计算机、机器人和能够理解世界的自动驾驶汽车的大脑。做前所未有的事情需要远见、创新和世界上最优秀的人才。作为 NVIDIA 的一员,您将沉浸在一个多元化、支持性的环境中,在这里每个人都受到激励,努力做出最好的工作。快来加入团队,看看您如何能够对世界产生持久的影响。 我们正在寻找关注性能分析和优化的工程师,帮助我们充分利用机器学习训练、推理和 NVIDIA AI 服务中的每一个时钟周期。我们正在全硬件/软件堆栈的各个层面工作,从 GPU 架构到机器学习框架,以实现最佳性能。这个角色提供了一个机会,可以直接影响一家快速成长的公司在硬件和软件路线图上的发展,该公司在多个领域引领 AI 革命。与世界级的软件工程师合作,实施极其快速的 SOTA 机器学习模型,帮助理解 NVIDIA 的 DL 软件和硬件堆栈的端到端性能。在拥有数百个 Petaflops 的强大企业级 GPU 集群上工作,并在全球任何人之前接触未发布的硬件。 您将要做的工作: • 执行探索性数据分析 (EDA) 并记录建议 • 具有多种 ML 框架的经验(PyT、Jupyter Notebooks、Pandas 等) • 能够对现有的预训练模型进行性能推理和微调 • 对 ML 算法和评估技术(如变换器、扩散、LSTM、SHAP)有基本了解 • 能够在标准工作负载管理平台(例如 MLFlow、SLURM)上编写脚本并运行工作负载 • 与 NVIDIA 的研究人员和工程师合作,提供支持以改善模型的设计、可用性和性能 我们希望看到的: • 2-4 年以上的机器学习模型实施和软件开发经验 • 计算机科学、计算机架构、数学、物理或相关技术领域的学士学位或同等经验 • 优秀的 Python 编程技能,至少对一个 ML 框架具有广泛的知识 • 较强的问题解决和分析能力 • 算法和机器学习基础知识 • 具有 Git 或其他源代码控制系统的经验 • 已发表的论文、专利或文章 NVIDIA 被广泛认为是科技界最受欢迎的雇主之一!我们拥有一些世界上最具前瞻性和勤奋的人在为我们工作。如果您富有创造力并且自主,我们希望听到您的声音! 您的基本薪资将根据您的地点、经验和类似职位员工的薪酬来确定。基本薪资范围为 108,000 美元 - 184,000 美元。 您还将有资格获得股票和福利。 此职位的申请将在 2025 年 12 月 14 日之前接受。NVIDIA 致力于营造多元化的工作环境,并自豪地成为一个平等机会的雇主。由于我们高度重视当前和未来员工的多样性,我们在招聘和晋升过程中不会基于种族、宗教、肤色、国籍、性别、性别表现、性取向、年龄、婚姻状况、退伍军人身份、残疾状态或任何其他受法律保护的特征进行歧视。 JR2009425