Head of AI Engineering
The ReWork Group
At The ReWork Group, we partner with high-growth startups and forward-thinking companies to build the future.
As the Head of AI Engineering, you'll own their AI research agenda, leading the design of foundation-model and reinforcement-learning systems that reason about simulations, generate hypotheses, and verify new scientific discoveries.
This is the rare seat where the science is the product, and the models you build are the instruments of discovery.
If you want to train frontier-grade models for scientific discovery, work at the seam of physics, AI, and large-scale simulation, and assemble a world-class team in a mission-driven, early-stage environment — this is the o ne.
What You'll Do:
- Set and drive the AI research roa dmap across foundation models, reinforcement learning, and agentic systems for scientific discovery.
- Contribute directly to core research — architect models, design training schemes, and prototype new agent behaviors alongside physicists and engineers.
- Recruit, hire, and manage AI researchers and engineers, building a high-bar, collaborative research culture.
- Be the internal thought leader o n AI — guiding technical trade-offs, priorities, and long-term architecture decisions.
You may be a good fit if:
- You b ring deep expertise in mode rn ML: foundation-model pretraining, RL (PPO/SAC-style methods, self-play, or similar), large-scale optimization, and distributed training across multi-GPU/multi-node systems.
- You've managed and grown small, high-performing research or engineering teams — even if you were never formally titled "Head of AI."
- You can set research direction , decompose ambitious goals into executable projects, and hold a high bar on both technical and communication standards.
- ...At The ReWork Group, we partner with high-growth startups and forward-thinking companies to build the future. As the Head of AI Engineering, you'll own their AI research agenda, leading the design of foundation-model and reinforcement-learning systems that reason about...Suggested
- A security technology company is seeking an Engineering Manager to lead its AI team. This role requires strong technical expertise in deep learning and computer vision, along with hands-on leadership experience. The ideal candidate will define the team's roadmap, architect...Suggested
$184k - $376k
...Director to lead the evolution of its core search experience using AI and machine learning. This role involves shaping how millions... ...The position requires over 10 years of experience in software engineering and/or machine learning, with a strong product mindset and...Suggested- An established industry player is seeking a Senior Director of Solutions Engineering to lead innovative teams in AI and high-performance computing solutions. This hybrid role offers the opportunity to shape enterprise solutions while collaborating with customers and partners...Suggested
- A leading AI-driven commerce solutions provider based in Mountain View, California, seeks an Engineering Leader to guide the incubation of next-generation SaaS solutions. The role demands hands-on leadership in AI engineering, requiring 8-10 years of backend software engineering...Suggested
$175.8k - $293k
...and data that power the world's largest organizations so they can seize a competitive advantage. We are looking for a Principal AI Engineer to build our next-generation Agentic AI platform from 0-1. This is a hands-on, delivery-driven role for a senior engineer who has...$185.2k - $299.48k
...Disruption, Collaboration, Execution, Integrity, and Inclusion. We weave AI into the fabric of everything we do and use it to augment the... ...need to do their best work. Your Career As a Principal AI Engineer for the Enterprise AI Platform, you will be a pivotal technical...Full timeWork at office$222k - $312.6k
Databricks is searching for a Senior Engineering Manager for Customer Experience Intelligence, responsible for shaping intelligent, AI-driven customer and Support experiences. The ideal candidate has over 10 years of experience in building customer workflows and leading...$272k - $431.25k
NVIDIA Gruppe is looking for a Principal Software Engineer to advance open-source AI inference. This hands-on role emphasizes running high-performance inference on NVIDIA platforms and involves collaboration across various teams. Key responsibilities include optimizing...- ...products that accelerate next-generation computing experiences-from AI and data centers, to PCs, gaming and embedded systems. Grounded... .... THE ROLE: AMD is looking for a Senior Staff AI Infra Engineer who is passionate about improving the performance of key...
$275.8k - $340.5k
...builds ML infrastructure designed to meet the unique demands of AI and ML innovation, supporting a wide range of use cases across teams... ...efficient ML experimentation, enhance the productivity of ML engineers, and drive the adoption of cutting-edge ML techniques. Our ML...Local areaRemote workWork from homeRelocationRelocation packageFlexible hours$148.7k - $297.3k
...place to work for diversity, working mothers, female executives, and scientists. THE OPPORTUNITY This Senior Principal AI/ML Engineer position can work out of our Santa Clara, CA location. The Principal ML Ops Engineer will lead the technical execution...Shift work$167k - $270.5k
Palo Alto Networks, Inc. is seeking a Technical Leader to develop AI applications within the GTM/CX domain. This role involves defining the architecture for scalable AI/ML systems and leading the design of intelligent agents. Ideal candidates will have 15+ years of experience...- A leading technology firm is seeking an experienced AI Engineering Manager to lead their AI/ML Platforms and Infrastructure team in Cupertino, California. In this role, you will mentor engineers, drive innovation in generative AI solutions and ensure scalability and reliability...
- NVIDIA Gruppe is seeking an AI Developer Technology Engineering Manager to lead a team focused on optimizing performance for real-world Deep Learning and Machine Learning applications. This role involves the development of algorithms to leverage Nvidia hardware effectively...
$141.8k - $258.6k
Apple Inc. seeks a Business Operations Engineering Program Manager for its Cupertino location. This role involves aligning operations with product milestones, managing budgets, and leveraging AI/ML opportunities for operational efficiencies. Ideal candidates have 3+ years...- A leading technology company in Santa Clara is seeking an Engineering Manager for their AI Developer Technology team. This role involves leading a skilled team to optimize algorithms for machine learning and deep learning applications. The ideal candidate should have substantial...
$296.3k
...builds ML infrastructure designed to meet the unique demands of AI and ML innovation, supporting a wide range of use cases across teams... ...efficient ML experimentation, enhance the productivity of ML engineers, and drive the adoption of cutting-edge ML techniques. Our ML...Local areaWork from homeFlexible hours- ...a common goal - crafting the best customer experiences. Description We are looking for a highly skilled and experienced AI Engineering Manager to lead the planning and execution for our next generation AIML Platforms, Infrastructure and AIOps for Channel Sales at...Worldwide
$272k - $431.25k
NVIDIA Corporation seeks a Principal AI and ML Infra Software Engineer in Santa Clara, California, to enhance the efficiency of AI/ML research on GPU Clusters. The role involves collaboration with various teams, monitoring infrastructure performance, and implementing improvements...$207k - $300k
Google Inc. is seeking a Software Engineering Manager II for Chrome AI in Mountain View, CA. This role involves managing and mentoring an engineering team, overseeing large-scale projects, and defining technical roadmaps while driving innovation in AI and machine learning...Full time- ...Head of Product and AI Engineering About the Company Expanding behavioral science & market research technology company Industry Market Research Type Privately Held About the Role The Company is in search of a Head of Product and AI Engineering to...
$275.8k - $340.5k
...builds ML infrastructure designed to meet the unique demands of AI and ML innovation, supporting a wide range of use cases across teams... ...efficient ML experimentation, enhance the productivity of ML engineers, and drive the adoption of cutting‑edge ML techniques. Position...Remote workRelocationRelocation packageFlexible hours$262.5k - $393.8k
Senior AI Engineering Manager - Channel Sales AIML Platform Cupertino, California, United States Machine Learning and AI Imagine the possibilities here. At Apple, great ideas swiftly transform into exceptional products, services, and customer experiences. We firmly believe...Relocation- A leading tech company located in Sunnyvale, California, is looking for a Director of Platform Engineering to architect and lead the development of its AI knowledge management platform. The role requires a minimum of 10 years of software engineering experience, with proven...
- NVIDIA Gruppe is seeking a Principal AI and ML Infra Software Engineer to join our Hardware Infrastructure team in Santa Clara, CA. In this role, you'll work closely with AI research teams to enhance efficiency by addressing infrastructure deficiencies for GPU Clusters,...
- ...Title: Prinicipal AI Engineer Location: Sunnyvale, California. Duration: 6 to 12+ Months Job Description: Skill Proficiency Why It's Needed Python (Advanced) Expert DSPy, FastAPI, async/await, MCP protocol LLM/Agent Systems...Contract work
- ...products that accelerate next-generation computing experiences-from AI and data centers, to PCs, gaming and embedded systems. Grounded... ..., we advance your career. THE ROLE: The AI Customer Engineering organization is looking for a Principal AI Systems Design...
- ...Head of Product and AI Engineering (CTO) About the Company Expanding market research technology company Industry Market Research Type Privately Held About the Role The Company is in search of a Head of Product and AI Engineering to spearhead...
- ...products that accelerate next-generation computing experiences-from AI and data centers, to PCs, gaming and embedded systems. Grounded... ...serves as the debug execution backbone of AMD's AI Customer Engineering organization, driving complex silicon, system, and fleet-level...
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Head of AI Engineering. Be the first to apply!
- principal infrastructure engineer Sunnyvale, CA
- principal data engineer Sunnyvale, CA
- chief engineer Sunnyvale, CA
- principal developer Sunnyvale, CA
- director data engineering Sunnyvale, CA
- general engineer Sunnyvale, CA
- senior chief engineer Sunnyvale, CA
- principal network engineer Sunnyvale, CA
- project engineer assistant project manager Sunnyvale, CA
- data center chief engineer Sunnyvale, CA


