Senior / Staff ML Training Optimization Engineer
$141k - $249kWaabi
Job Description Job Description Waabi, founded by AI visionary Raquel Urtasun, is the leader in Physical AI. With a world-class team, we're unlocking the next era of autonomous transportation with technology that's powering commercial autonomous trucks and robotaxis. Waabi is backed by and partners with world leaders in AI, automotive, logistics, and deep tech. With offices in Toronto, San Francisco, Dallas, and Pittsburgh, Waabi is growing quickly and looking for diverse, innovative and collaborative candidates who want to impact the world in a positive way. To learn more visit: You will... - Build standardized distributed training frameworks for research and production, drive our training towards new levels of stability and efficiency. - Comprehensively profile model runtime and memory to pinpoint performance bottlenecks. - Identify and evaluate emerging technologies that can be adopted into Waabi’s training and inference frameworks. Examples include designing new CUDA kernels, quantization-aware training and inference, and compilation/deployment techniques. - Work with researchers and ML engineers on best-practices for optimal resource usage. - Create and improve tooling and dashboards to ensure broad adoption of your work. Qualifications: - MS/PhD or Bachelors degree with a minimum of 4 years of industry experience in Computer Science, Robotics and/or similar technical field(s) of study. - Solid coding proficiency in a variety of coding languages including Python, C++ or Rust. - Experience in deep learning frameworks such as PyTorch or Jax. - Skilled in profiling CPU and GPU code using tools such as PyTorch Profiler and NVIDIA Nsight. - Open-minded and collaborative team player with willingness to help others. - Passionate about self-driving technologies, solving hard problems, and creating innovative solutions. Bonus/nice to have: - Experience in identifying when custom CUDA kernels are needed, and implementing them. - Experience in Bazel in a monorepo environment, and integrating third party packages into dev environments. - Experience with Kubernetes-based training platforms. The US yearly salary range for this role is: $141,000 - $249,000 in addition to competitive perks & benefits. Waabi’s yearly salary ranges are determined based on several factors in accordance with the Company’s compensation practices. The salary base range is reflective of the minimum and maximum target for new hire salaries for the position across all US locations. Note: The Company provides additional compensation for employees in this role, including equity incentive awards and an annual performance bonus. Perks/Benefits: Waabi provides a competitive benefits package that includes: - Competitive compensation and equity awards. - Health and Wellness benefits encompassing Medical, Dental and Vision coverage. - Unlimited Vacation. - Flexible hours and Work from Home support. - Daily drinks, snacks and catered meals (when in office). - Regularly scheduled team building activities and social events both on-site, off-site & virtually. - World-class facility that includes a gym, games room (ping pong table, video game consoles, board games, etc), multiple collaborative working spaces and a gorgeous patio!(when in office) - As we grow, this list continues to evolve! Waabi is an equal opportunity employer that celebrates diversity and is committed to creating a supportive, inclusive, and accessible environment for all employees. If reasonable accommodation is needed to participate in the job application or interview process please let our recruiting team know. We may use artificial intelligence (AI) tools to support parts of the hiring process, such as reviewing applications, analyzing resumes, or assessing responses. These tools assist our recruitment team but do not replace human judgment. Final hiring decisions are ultimately made by humans. If you would like more information about how your data is processed, please contact us.
- ...that matter. Autodesk is seeking a Senior ML Engineer, ML Systems and Infrastructure to design... ...-scale data pipelines, distributed training systems, evaluation frameworks, and production... ...semi-structured technical datasets Optimize distributed workflows for performance,...SeniorTrainingTemporary workFor contractorsRemote work
- ...data scientists and AI engineers, driving the end-to-... ...advanced simulation and optimization techniques, including... ...collective bar by mentoring Senior-level engineers and... ...core architectures, training methodologies, and... ...technical lead for complex AI/ML/DS projects, taking...SeniorTrainingFull timeWork at office
$178.88k - $320.65k
...Job Requisition ID # 26WD94803 Senior Principal Machine Learning Engineer, ML Platform and Systems Architecture Position Overview The work we do... ...that enable machine learning across Autodesk, including training infrastructure, data platforms, evaluation and...SeniorTrainingFor contractorsRemote work$68.9k - $98.2k
...Job Description Bloom Engineering, a Wabtec Company, improves... ...seeking a highly experienced Senior Field Service Engineer to support... ..., maintenance, and optimization of new, existing, and retrofitted... ...10-Hour and NFPA 70E safety training (or ability to obtain)...SeniorTrainingRemote jobFor contractorsWork experience placementWork at officeLocal areaWorldwide$136.5k - $225k
...Description Job Description You're an engineer who is excited to work on cutting-edge... ...background and interests, you may also help optimize and deploy models to the vehicle,... ...techniques Experience building, training, or evaluating neural networks Proficiency...SeniorTrainingInternship$106.8k - $194.8k
...working world. WAF Operations Solution Engineer PRACTICE DESCRIPTION: As a WAF... ...of web application firewalls to maintain optimal performance and security for client... ...growth. You will have access to continuous training and mentorship, enabling you to enhance...SeniorTrainingSummer holidayFlexible hours- ...seeking a future team member for the role of Cryptography PKI Engineer to join our Cybersecurity team. This role is located in Lake Mary... ..., and continuous learning. We offer quality career resources, training, certifications, development opportunities, and a comprehensive...SeniorTrainingWorldwide
$160k - $210k
...Hardware Infrastructure & Ci/cd Engineer San Jose, CA OR Pittsburgh,... ...programming languages and AI/ML frameworks. This level of... ...Database Operations: Manage and optimize PostgreSQL (operations data store... ...guides, and runbooks; provide training and office hours to help teams...TrainingWork at office$104k - $210k
...Forward Deployed Engineer At BNY, our culture allows us to run our company better and... ...in active learning, complete mandatory training, and stay abreast of emerging technologies... ...in ambiguity and driving discovery with senior stakeholders. Experience balancing speed...SeniorTrainingTemporary workWork experience placementWorldwideFlexible hours$128.6k - $192.9k
...design practice. Our architects, engineers, interior designers,... ...Specialty Services group is seeking a Senior Leader to join and guide our... ...development and mentoring of our staff. We are constantly expanding... ...external clients Mentor and train entry level staff in both...SeniorTrainingFull timeContract workFor contractorsWork at office- ...Job Description Title: Sr. Engineer - UI Reports To: Associate Manager - Engineering... ...we embrace self expression, culture, optimism and connection. Through the brand... ...recruitment, hiring, compensation, assignment, training, promotion, performance evaluation,...SeniorTrainingFull timePart timeSummer workCasual workLocal area
$48.08 - $76.52 per hour
...currently hiring for a regular Full-Time Staff/Sr. Medical Dosimetrist to support the... ...is responsible for generating clinically optimal treatment plans for radiation therapy patients... ...such as providing instruction and training to new staff members, physician residents...SeniorTrainingFull timeTraineeshipCurrently hiringLocal areaWork from homeMonday to Friday- ...STS USA, Inc., headquartered in Pittsburgh, PA, seeks a Senior Quality Assurance Engineer responsible for implementing quality management processes... ...of Work/Life Programs such as Flextime and a variety of Training and Development opportunities. Equal Opportunity...SeniorTrainingTemporary workRelocationFlexible hours
- ...Job Description Position Summary The Senior Engineer supports the daily operation,... ...Facilities - Coordinate with maintenance staff, contractors, and vendors - Provide technical... ...and troubleshooting - Assist in training and developing junior staff Life...SeniorTrainingFor contractorsWork at officeLocal area
- ...advancement ~ Paid time off ~ Training & development ~ Vision insurance... ...Job Summary We are seeking a Senior Civil/Environmental Engineer (PE preferred) to join our team! We... ...a team atmosphere and mentor newer staff. Responsibilities Ability to...SeniorTrainingWork from homeFlexible hours
$105.8k - $174.8k
...and Decision Science – AI Native Engineering Physical AI Engineering Consultant, Senior Consultant The opportunity... ...pipelines to ensure data integrity and optimize learning processes, all while... ...simulations, and experience with popular ML packages such as TensorFlow,...SeniorFull timeWork experience placementSummer holidayFlexible hours$104.8k - $149.3k
...industry. Our products support rail customer operations for positive train control, dispatch systems, SCADA, and customer information... ...data analytics. How will you make a difference? As a Senior Engineer in the Surface Transportation Group, you will be responsible for...SeniorTrainingWork experience placementWork at officeRemote workWorldwide$172k - $229k
...Senior Machine Learning Engineer, Data Mining Boston, MA February 10, 2026 At... ...matter most. Omnitag, our ML-powered multimodal data mining... ...learning, retrieval optimization, and reasoning systems. Your... ...for teams working on post-training analysis, error diagnosis,...SeniorTrainingWork at officeRemote work$160k - $210k
...Senior Software Optimization Engineer Efficient is developing the world's most energy-efficient general-purpose computer processor. Efficient's patented... ...using standard high-level programming languages and AI/ML frameworks. This level of efficiency makes perpetual,...SeniorWork experience placementImmediate start- ...Senior Software Engineer - Scalable Machine Learning Latitude AI develops automated... ...data is vital to achieve optimal performance and to enable a... .... In the Scalable ML team, we are looking at methods... ...improve the efficiency of model training, enabling ML developers to work...SeniorTrainingWork at officeImmediate start
- ...Senior Sales & Proposal Engineer The successful Senior Process Engineer will be responsible for marketing and selling all of the company's steelmaking... ...vacation, paid holiday time, 401k with a company match, training, a tuition reimbursement program and more! SMS group is...SeniorTraining
$146k - $234k
...more efficient and accessible for all. We're searching for a Senior Software Engineer - Camera Systems. In this role, you will Develop... ...skills, experience, qualifications, relevant education or training, and market conditions. These ranges may be modified in the...SeniorTrainingWork at officeLocal area3 days per week$106.2k - $166k
...Nuclear Laboratory is seeking a Data Engineer to play a key role in advancing... ...and lifecycle management of AI/ML models and agents in production. Optimize data workflows and systems... ...relevant experience; or Currently a Senior Reactor Training Engineer, Refueling Engineer, or...SeniorTrainingLocal areaFlexible hours$106.9k - $176.5k
...and Decision Science – AI Native Engineering AI/Machine Learning Engineer, Senior Consultant The opportunity... ...pipelines to ensure data integrity and optimize learning processes, all while... ...-impact, high-value production AI/ML solutions on a major cloud platform...SeniorFull timeWork experience placementSummer holidayFlexible hours$179.2k - $268.8k
...operations, systems and safety engineering - all dedicated to making a... ...road, ensuring that they are optimized for onboard performance and... ...cutting-edge tracking algorithms, ML algorithms, and models that... ..., experience, relevant training and education, and skill level...SeniorTrainingPermanent employmentFull timeWork at officeImmediate startVisa sponsorship$149k - $198.5k
...autonomy software. We are seeking a talented senior engineer to contribute to evaluation of our... ...performance, and provide meaningful insight to ML developers and teams, working closely... ...). Assess coverage of existing training and testing, then curate test sets that...SeniorTraining$179.2k - $268.8k
...operations, systems and safety engineering - all dedicated to making a... .... This work helps engineers train models, test software, and move... ...and monitor large numbers of ML jobs used across the company... ...are reliable and being optimally used Build with scale and...SeniorTrainingPermanent employmentFull timeWork at officeImmediate startVisa sponsorship- ...Sr Software Engineer - Transportation Systems American Eagle is a youth culture brand... ...jeans–we embrace self expression, culture, optimism and connection. Through the brand... ...recruitment, hiring, compensation, assignment, training, promotion, performance evaluation,...SeniorTrainingFull timePart timeSummer workCasual workLocal area
$179.2k - $268.8k
...operations, systems and safety engineering - all dedicated to making a... ...take advantage of hardware optimizations. You will play a key role... ...transformer, etc...) and classical ML (Random Forest, SVM, etc..)... ..., experience, relevant training and education, and skill level...SeniorTrainingPermanent employmentFull timeWork at officeImmediate startVisa sponsorship$144k - $192k
...looking for a Machine Learning Systems Engineer to join our ML Acceleration team. In this role, you... ...that enable our researchers to train frontier models at scale, focusing obsessively... ...doing: Performance Profiling & Optimization : Utilize profiling tools (e.g.,...TrainingFull timeWork at officeRemote work
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Senior / Staff ML Training Optimization Engineer. Be the first to apply!
- assistant engineer Pittsburgh, PA
- engineering aide Pittsburgh, PA
- staff engineer Pittsburgh, PA
- technology administrator Pittsburgh, PA
- senior staff systems engineer Pittsburgh, PA
- machine learning engineer Pittsburgh, PA
- senior ml engineer Pittsburgh, PA
- computer vision machine learning engineer Pittsburgh, PA
- senior platform engineer Pittsburgh, PA
- senior procurement Pittsburgh, PA



