Jobs in the United States

25+ jobs

Save this search

Added 19 mins ago
Senior Staff Software Engineer, AI Telemetry and Developer Intelligence
Google
San Jose, CA
7 years of experience leading technical project strategy, ML design, and end-to-end Machine Learning (e.g., model deployment, model evaluation, data processing, debugging).
Added 19 mins ago
Senior Software Engineer, ML Performance, GenAI, DeepMind
DeepMind
Mountain View, CA
At Google DeepMind our mission is to build the world's first general-purpose learning agent. Central to this mission is the complex task of measuring the intelligence of our prototypes. As…
Added 22 mins ago
CP
Senior Software Engineer
Comprehend PT Inc.
BUFFALO, NY
- $120,000 USD / year
Added 23 mins ago
Senior Software Engineer, Prime Video Personalization and Discovery
Amazon
New York, NY
Prime Video Personalization and Discovery (PVPD) is seeking a Senior Software Development Engineer to lead Personalization and Discovery customer experiences. This role will work across…
Added 23 mins ago
Senior Research Engineer, GenMedia Safety, DeepMind
Google
New York, NY
Share Senior Research Engineer, GenMedia Safety, DeepMind
Added 31 mins ago
OS
Senior Splunk ITSI Engineer
Olympus Solutions Inc.
Washington, DC
- $165,000 to $195,000 USD / year
Olympus Solutions, Inc. is seeking a Full-Time, highly motivated Senior Splunk ITSI Engineer to own the service intelligence layer of a federal customer’s Splunk environment. In this…
Added 32 mins ago
J
Staff Machine Learning Engineer - Team Lead (Active Secret Clearance)
Jobgether
Remote, United States
- $230,000 to $280,000 USD / year
…applications and next steps. Our partner is looking for a Staff Machine Learning Engineer - Team Lead (Active Secret Clearance) based in the United…
Added 32 mins ago
Senior Staff Software Engineer
Jobgether
Remote, United States
This position is listed on behalf of a partner company, who manages all applications and next steps. Our partner is looking for a Senior Staff Software Engineer based in the United States.
Added 2 hours ago
Senior Software Engineer, Agentic Automations, Connect Sales AI, Google Ads
Google
Mountain View, CA
3 years of experience with end-to-end machine learning (e.g., model deployment, model evaluation, data processing, debugging).
Added 2 hours ago
Senior Software Engineer, Machine Learning, Discover Interest Modeling
Google
Mountain View, CA
5 years of experience with machine learning algorithms and tools (e.g., TensorFlow), Recommender Systems, deep learning, or personalization.
Added 3 hours ago
Senior Staff Software Engineer, AI/ML, Multimodal Rich Features
Google
Mountain View, CA
5 years of experience leading design for machine learning products and optimizing systems that incorporate machine learning.
Added 3 hours ago
Senior Software Engineering Manager, AI/ML
Google
San Francisco, CA
7 years of experience leading technical project strategy, ML design and optimizing end-to-end Machine Learning (e.g., model deployment, model evaluation, data processing, debugging, fine…
Added 4 hours ago
Global AI Transformation & Solutions Lead
AFRY
Appleton, WI
…across areas such as MES, digital twins, process modelling, machine learning and industrial…
Added 4 hours ago
J
Technical Architect Senior
Jobgether
Remote, United States
- $201,950 USD / year
…capabilities, actuarial modeling environments, and end-to-end machine learning development and…
Added 4 hours ago
Senior Full Stack Java Engineer - Workforce Management (WFM)
Miratech
Remote, United States
We are looking for a Senior Full Stack Engineer to join the Workforce Management product stream and build enterprise-scale solution supporting forecasting, capacity planning, scheduling,…
Added 4 hours ago
Senior Full Stack Java Engineer - Workforce Management (WFM)
Miratech
Remote, United States
We are looking for a Senior Full Stack Engineer to join the Workforce Management product stream and build enterprise-scale solution supporting forecasting, capacity planning, scheduling,…
Added 5 hours ago
J
Senior Machine Learning Engineer, Inference
Jobgether
Remote, United States
- $250,000 to $300,000 USD / year
…all applications and next steps. Our partner is looking for a Senior Machine Learning Engineer, Inference based in the United…
Added 5 hours ago
DO
Engineer, Water Resources
Department of Water Resources
CA
- $6,488 to $12,152 USD / month
Under the supervision of a Senior Engineer, Water Resources in the Central Valley Modeling Section of the Modeling Support Office, is responsible for developing and maintaining present and…
Added 5 hours ago
Consultant, Data Engineer - Reference Data Management
AB3 Nationwide Life and Annuity Insurance Co
OH · AZ
…visualization, governance, engineering, AI engineering, and machine learning to produce targeted data solutions. If you have the drive and desire to be part of a future-forward data-enabled…
Added 6 hours ago
Staff Engineer - Software Engineering
Capital One
Cambridge, MA
You will work alongside our talented team of developers, machine learning experts, product managers and people leaders. Our Distinguished Engineers are leading experts in their domains,…
Added 6 hours ago
Machine Perception (MP) Engineer – Level 6
Northrop Grumman
Gilbert, AZ
…bring your pioneering spirit to our collaborative teams. As a Machine Perception (MP) Engineer – Level 6 located in Dulles, VA, or Gilbert, AZ , you’ll be a linchpin in missions of…
Added 6 hours ago
Lead Application Engineer
Deliberately.ai
San Francisco, CA
We are looking for a Lead Application Engineer to join our team and lead the development of key projects across our application. This senior-level role requires someone who can…
Added 7 hours ago
Senior/Staff Software Engineer [Full Stack]
Prudentia Sciences
Boston, MA · New York, NY
We're seeking an exceptional Senior/Staff Software Engineer to build and lead our core platform as we scale. This is a rare opportunity to join a high-impact technology company in a major…
Added 7 hours ago
Senior Quantum Platform Engineer
Quanta Standard Inc
Washington, DC · Arlington, VA · Alexandria, VA
Quanta Standard Inc. is seeking a highly skilled and experienced Senior Quantum Platform Engineer to join our team. This role requires a minimum of 8 years of experience in similar roles…
Added 7 hours ago
AR
Mid/Senior Level Robotics Software Engineer
AlphaOne Robotics
Dallas, TX
- $100,000 to $150,000 USD / year
Knowledge of machine learning frameworks (e.g., PyTorch, TensorFlow) and models (eg. Yolo) for robotics applications like perception or decision-making.

DeepMind

Senior Software Engineer, ML Performance, GenAI, DeepMind

Mountain View, CA, United States

Apply About employer

Applications open in a new tab.

Job details

Posting language: English

Original job description

Minimum qualifications:

Bachelor's degree in Computer Science, Software Engineering, Mathematics, or a related STEM field or equivalent practical experience.
5 years of experience in software development, systems programming, or applied ML engineering or equivalent education credit (e.g., internships, student research, pre-graduate working experience).
3 years of experiencewith performance optimization, systems data analysis, visualization tools, or debugging.
3 years of experience with Generative AI, Large Language Models (LLMs), Machine Learning and related frameworks.
1 year of experience designing and launching production software products or foundation AI systems.

Preferred qualifications:

Master's or PhD in Computer Science, Artificial Intelligence, Robotics, or related technical field.
3 years of experience in a technical leadership role setting architectural direction and guiding project teams.
Experience developing, evaluating, or serving large language models, agentic workflows, multimodal pipelines, or AI runtime APIs (e.g., PyTorch, JAX, Triton, vLLM, Gemini API).
Expertise or strong interest in the following areas: performance debugging and optimization of complex workloads, design of performance tools, compiler design and code optimization, high-performance software development techniques, concurrent programming, or multi-core computer architectures.

At Google DeepMind our mission is to build the world's first general-purpose learning agent. Central to this mission is the complex task of measuring the intelligence of our prototypes. As a Software Engineer, you will be working with the cutting edge AI agents developed by our exceptional team of Machine Learning and Neuroscience research scientists. Your responsibilities will include everything from creating systems for agent testing using 2D and 3D games to developing test problems within physics simulators. You will create graphical visualization of results, build competitive agent leaderboards and test new algorithms on robots. To succeed in this role you will need to have a strong foundation in software engineering and enjoy working on a wide range of challenging problems within a mission-driven team.

We focus on identifying and achieving performance improvements in production and research ML workloads, with an emphasis on LLM inference - including serving latency, throughput, and cost per token - as well as the emerging demands of agentic workflows on current and next-generation TPU platforms. We work across the entire stack, ranging from model architecture and sharding strategies to Pallas/Mosaic kernels and XLA. Additionally, we partner closely with the Gemini modeling, Deployment, Compiler, and TPU hardware teams to shape the capabilities of next-generation accelerators. Artificial intelligence will be one of humanity’s most transformative inventions. At Google DeepMind, we are a pioneering AI lab with exceptional interdisciplinary teams focused on advancing AI development to solve complex global challenges and accelerate high-quality product innovation for billions of users. We use our technologies for widespread public benefit and scientific discovery, ensuring safety and ethics are always our highest priority.

We are pushing the boundaries across multiple domains. Our global teams offer diverse learning opportunities and varied career pathways for those driven to achieve exceptional results through collective effort. Individual pay is determined by factors including job-related skills, experience, and relevant education or training.

US: $174000 - $252000 (USD) + 15% bonus target + equity + benefits

Learn more about benefits at Google.

Drive down the cost per token of Gemini inference by optimizing throughput, accelerator utilization, batching, scheduling, and KV cache efficiency. Achieve these gains through expert optimization and by building agentic workflows that enable AI agents to discover, implement, and validate performance improvements at scale.
Be among the first to bring Gemini to new TPU generations by characterizing hardware at launch, tuning kernels, sharding, and serving configurations for specific memory hierarchies and interconnects, and delivering production-ready performance. Share these insights to influence future TPU hardware and compiler roadmaps.
Implement model compression techniques - such as quantization and sparsity - to reduce the cost-per-token for production Gemini models. Evaluate new architectures, optimizers, and training methods to achieve target quality with reduced compute requirements.

Apply About employer

Report this job