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.
- 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
- 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
$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...Suggested$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...SuggestedFull timeWork at office$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- ...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...
$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...$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- 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...
$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 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$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...- ...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
- ...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...
- 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,...
- ...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...
$147k - $237.5k
Job Summary Design and develop testing frameworks for AI/ML models and LLM applications. Build automated pipelines for model validation... ...validation, and hallucination detection. Collaborate with engineers, and product teams to define test strategies. Monitor model performance...- ...Title: Python AI Engineer Location: Sunnyvale, CA Salary Range: $50-55/hr on W2 Required Skills & Qualifications ~7 years of experience with Python, Machine Learning, and Generative AI. ~ Strong expertise in Python and core Machine Learning techniques...Permanent employmentWork experience placementImmediate start
- Palo Alto Networks, Inc. is looking for a skilled engineer to design and develop testing frameworks for AI/ML models and LLM applications in Santa Clara, California. The role involves building automated pipelines for validation and testing, collaborating with product teams...
$178k - $288k
Palo Alto Networks, Inc. is seeking a Sr. Engineering Manager for their Sales Cloud portfolio in Santa Clara, CA. This position involves redefining how the sales organization operates by integrating AI and modern frameworks into Salesforce. The ideal candidate will have...Work at office$272k - $431.25k
NVIDIA Gruppe is seeking expert engineers to develop next-generation scaling AI supercomputing platforms. Candidates should possess extensive coding experience, particularly in C/C++ and Python, with a strong understanding of system architecture and time series databases...- 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...
$140k - $170k
...AI/Python Software Engineer (Onsite, Austin TX or Sunnyvale CA) Client: Apple Location: Onsite, Austin TX or Sunnyvale CA Salary: $140k - $170k Recruit JC: 118144 Key Responsibilities: Design and implement AI/ML models leveraging frameworks like TensorFlow...- A leading automotive company is seeking a Principal AI Engineer to lead the design and optimization of its AI platform. The successful candidate will guide the infrastructure for large-scale training and cloud inference, working closely with data scientists and engineers...Remote job
- NVIDIA Gruppe in Santa Clara is seeking an experienced manager to lead a solution engineering team focused on customer support of NVIDIA software products like AI Enterprise and Omniverse. The role demands strong problem-solving abilities and excellent communication skills...
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Head of AI Engineering. Be the first to apply!
- chief design engineer Santa Clara, CA
- principal infrastructure engineer Santa Clara, CA
- principal data engineer Santa Clara, CA
- chief engineer Santa Clara, CA
- principal developer Santa Clara, CA
- director data engineering Santa Clara, CA
- general engineer Santa Clara, CA
- senior chief engineer Santa Clara, CA
- principal network engineer Santa Clara, CA
- data center chief engineer Santa Clara, CA

