Waabi

Waabi is an innovative leader in artificial intelligence, dedicated to revolutionizing autonomous vehicle technology. Founded with a mission to create safer and

Senior - Staff ML Training Optimization Engineer

Location

Canada

Posted

2 days ago

Salary

$141K - $249K / year

Seniority

Senior

Job Description

Senior - Staff ML Training Optimization Engineer

Waabi

Senior - Staff ML Training Optimization Engineer Location: Remote, Canada, United States Job Description: Remote US & Canada / Dallas, TX / Phoenix, AZ / Pittsburgh, PA / San Francisco, CA / Toronto, ON Software – Autonomy & Algorithms / Full-time / Hybrid 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. 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 and identifying potential inconsistencies or verification signals in application materials based on available information. 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.

Related Categories

Related Job Pages

More Engineer Jobs

EPAM Systems logo

Senior Full-Stack Engineer - Python and Angular

EPAM Systems

EPAM Systems is an information technology (IT) company that has become a leading global digital and product design, digital platform engineering, and product de

Engineer2 days ago

Title: Senior Full-Stack Engineer (Python & Angular) Location: Remote in Armenia/ Kazakhstan/ Kyrgyzstan/ Uzbekistan/ Georgia Python.Web Job Description: We are seeking a Senior Full-Stack Engineer (Python & Angular) to contribute to the development of a high-load enterprise-grade platform in the Oil & Gas domain. This role primarily focuses on backend development (~70% Python) while also involving frontend responsibilities with Angular and participation in system architecture and design. This is a fully remote position that offers you the flexibility to work from any location in Armenia, whether it's your home or well-equipped offices in Yerevan or Gyumri. Responsibilities - Develop Python-based microservices and ensure their scalability and performance - Contribute to frontend development using Angular where necessary - Design APIs and integrate distributed systems effectively - Take part in critical architecture and technical decision-making processes - Operate within Docker environments and manage CI/CD pipelines - Work in cloud-based setups to support platform deployment - Collaborate with a cross-functional Agile team to achieve project goals Requirements - 3+ years of experience with Python in backend development - Proficiency in Angular or similar frontend frameworks - Solid understanding of microservices architecture, API design and OOP & SOLID principles - Expertise in Docker, Git and Azure DevOps - Strong problem-solving capabilities and a proactive approach to ownership - English proficiency at B2 level or higher Nice to have - Background in Kubernetes - Familiarity with messaging systems like RabbitMQ and event-driven architecture - Qualifications in MongoDB or other NoSQL databases - Experience with Azure Cloud systems We offer/Benefits We connect like-minded people - Delivering innovative solutions to industry leaders, making a global impact   - Enjoyable working environment, whether it is the vibrant office or the comfort of your home  - Opportunity to work abroad for up to two months per year  - Relocation opportunities within our offices in 55+ countries - Corporate and social events    We invest in your growth - Leadership development, career advising, soft skills and well-being programs  - Certifications, including GCP, Azure and AWS</li> - Unlimited access to LinkedIn Learning and Udemy - Free English classes with certified teachers     We cover it all - Participation in the Employee Stock Purchase Plan  - Monetary bonuses for engaging in the referral program  - Comprehensive medical & family care package    - Four trust days per year for personal needs - Discounts for fitness clubs - Benefits package (hotels, restaurants, stores and services) EPAM Armenia is a team of talented innovators united by a passion for technology. In 2014, we opened our first office in Yerevan, and now we have a second engineering hub in Gyumri. We've built a continuously learning organization that helps its employees rapidly advance their careers. Here you will work with the world's industry leaders, support impactful projects using the latest technologies, collaborate with multi-national teams, and have access to a wide variety of development opportunities.

Armenia + 4 moreAll locations: Armenia | Kazakhstan | Kyrgyzstan | Uzbekistan | Georgia

Role Description As a remote Senior Information Assurance (IA) Engineer, you will lead critical cloud security operations, DevSecOps pipeline integration, and compliance auditing. The ideal candidate will have extensive hands-on experience securing AWS cloud infrastructure operating at DoD Impact Level 5 (IL5), with preparations for Impact Level 6 (IL6). This senior role requires deep expertise in Application Security (AppSec) tools, secure code methodologies, and vulnerability management, along with a foundational understanding of the DoD Risk Management Framework (RMF) to support continuous compliance and Authority to Operate (ATO) efforts. Responsibilities - Lead the design, implementation, and maintenance of advanced security controls within AWS cloud environments operating at DoD IL5, ensuring architectural readiness for future IL6 integration. - Architect and manage security testing integrations within Azure DevOps (ADO) and Git environments. Embed and automate Application Security (AppSec) into the CI/CD pipeline. - Administer, utilize, and optimize security testing tools, specifically Fortify and Tenable, to identify, assess, and drive the remediation of code and infrastructure vulnerabilities. - Conduct continuous compliance auditing and security assessments. Apply foundational (junior-level) knowledge of the DoD Risk Management Framework (RMF) to support the generation and maintenance of artifacts required for Authority to Operate (ATO). - Analyze complex security scan results, coordinate cross-functionally with development teams to remediate findings, and enforce strict DoD security configuration guidelines (e.g., STIGs, SRGs). - Provide input to internal project teams consisting of both contract partners and government customers to ensure effective use of IT solutions, tools, and processes. Qualifications - MA/MS in a related field with a minimum of 10+ years of relevant experience, or BA/BS in a related field with a minimum of 15+ years of relevant experience. - Five years’ experience in monitoring, testing, troubleshooting, and performing corrective actions for Cybersecurity IT systems. - Active Secret security clearance. - U.S. citizenship required. - CompTIA Security+ Certification. - Direct operational experience with security scanning and vulnerability management tools, specifically Fortify and Amazon Inspector. - Proficiency with Git and CI/CD orchestration using Azure DevOps (ADO). - Demonstrates foundational knowledge of the RMF lifecycle, ATO processes, and DoD security methodologies. - Practical understanding of key technologies including defensive cyber event logging, application security, commercial DoD PKI, and AWS/Azure cloud authorization. - Experience in developing metrics from associated Cybersecurity elements, including trend analysis. - Ability to provide input to internal project teams consisting of both system users and IT resources as well as functional proponents and Army communities. Preferred Qualifications - Prior experience working with ECMA to implement DoD cloud solutions. - Familiarity with additional code quality and security tools such as SonarQube, Semgrep, or Checkmarx. - Prior experience transitioning enterprise systems from DoD IL5 to IL6 environments. - Advanced cloud or security certifications (e.g., AWS Certified Security Specialty, CISSP, CCSP). Benefits - A competitive compensation package. - Comprehensive health and wellness benefits, including medical, dental, and vision plans. - Access to company-provided retirement savings options with matching contributions. - Opportunities for professional growth and continued learning. - Additional perks such as discounts on various services and products. - Remote work.

United States
EPAM Systems logo

Senior Full-Stack Engineer

EPAM Systems

EPAM Systems is an information technology (IT) company that has become a leading global digital and product design, digital platform engineering, and product de

Engineer2 days ago

Senior Full-Stack Engineer (AI-Enabled) Location: Remote in Georgia Job Description: JavaScript (Fullstack) FacebookLinkedInSend via email Looking for something else? Find a vacancy that works for you. Send us your CV to receive a personalized offer. We are seeking a Senior Full-Stack Engineer with strong hands-on experience in modern web development and practical exposure to AI technologies. You will build scalable applications and contribute to integrating AI-driven features into production systems. Responsibilities - Build and maintain full-stack applications using TypeScript, React and Node.js - Develop APIs (REST, tRPC, GraphQL) and manage database schemas (PostgreSQL) - Contribute to AI feature development, including LLM integrations and workflow automation - Collaborate on system design, architecture and implementation decisions - Ensure code quality through testing, reviews and best practices - Work in Agile teams and deliver production-ready features Requirements - 5+ years of full-stack development experience - Strong proficiency in TypeScript, React, Node.js - Experience with modern frameworks (Next.js, Tailwind, headless UI) - Experience with relational databases and ORMs (Prisma, Drizzle) - Hands-on experience with AWS and CI/CD pipelines - Experience with Docker - Practical experience integrating AI APIs (e.g. OpenAI) into applications - Familiarity with AI-assisted development tools (Copilot, Cursor, etc.) - Understanding of AI SDLC Nice to have - Experience with microfrontends and Single-SPA - Testing frameworks (Vitest, Playwright) - Kubernetes or infrastructure-as-code tools.

Georgia
NBCUniversal logo

Engineer, Builder Enablement

NBCUniversal

Here you can create the extraordinary. Join us.

Engineer2 days ago
Full TimeRemoteTeam 10,001+Since 2004H1B Sponsor

• Identify common design and implementation needs across application engineering teams • Create and maintain high-quality documentation, reference architectures, and implementation guides • Develop reusable artifacts such as Terraform or similar modules, code templates, and example implementations • Partner with enterprise platform and foundation teams • Collaborate with architects to translate architectural principles • Contribute to and evolve AI-assisted developer workflows • Evaluate and choose the right medium for enablement • Gather feedback from application teams and iterate on paved paths • Advocate for a developer-centric approach to platform adoption

New York
$140K - $165K / year