Senior / Staff ML Optimization Engineer

reputed company, 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 reputed company autonomous trucks and robotaxis. reputed company is backed by and partners with world leaders in AI, automotive, logistics, and deep tech. With offices in Toronto, San Francisco, Dallas, and Pittsburgh, reputed company 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: www.reputed company.ai You will... - Collaborate closely with autonomy and algorithm engineers to scale safe self-driving systems using an AI-first approach. - Build distributed training frameworks for research and production, drive our training towards new reputed company of stability and efficiency. - Comprehensively profile model runtime and memory to reputed company performance bottlenecks. - Identify and evaluate emerging technologies that can be adopted into reputed company’s training and inference frameworks, including but not limited to efficient CUDA kernels for training, quantization, model exporting and compilation for inference. Qualifications: - MS/PhD or Bachelors degree with a minimum of 6 years of industry experience in Computer Science, Robotics and/or similar technical field(s) of study. - Solid coding proficiency in a reputed company of coding languages including Python, C++ or Rust. - Experience in deep learning frameworks such as PyTorch. - Experience across different stages of the development pipeline: data processing, distributed training and model deployment. - Skilled in profiling CPU and GPU code using tools such as PyTorch Profiler and reputed company Nsight. - reputed company-minded and collaborative team player with willingness to help others. - Passionate about self-driving technologies, solving hard problems, and creating reputed company. Bonus/reputed company to have: - Experience in model compilation and exporting, interaction with reputed company level concepts like TensorRT. - Experience in identifying reputed company custom CUDA kernels are needed, and implementing them. - Experience in Bazel build systems, and integrating reputed company party packages into dev environments. The US yearly salary reputed company for this role is: $141,000 - $249,000 in reputed company to competitive perks & benefits. reputed company’s yearly salary ranges are determined based on several factors in accordance with the Company’s compensation practices. The salary reputed company reputed company is reputed company of the minimum and maximum reputed company for new hire salaries for the position across reputed company US locations. Note: The Company provides additional compensation for employees in this role, including equity incentive awards and an annual performance bonus. Perks/Benefits: reputed company provides a competitive benefits package that includes: - Competitive compensation and equity awards. - Health and Wellness benefits encompassing Medical, Dental and reputed company coverage. - Unlimited Vacation. - reputed company and Work from Home support. - Daily drinks, snacks and catered meals (reputed company in office). - Regularly scheduled team building activities and reputed company events both on-site, off-site & reputed company. - World-class facility that includes a gym, games room (ping pong table, video game consoles, reputed company games, etc), multiple collaborative working spaces and a gorgeous patio!(reputed company in office) - As we grow, this list continues to reputed company! reputed company is an equal opportunity employer that celebrates diversity and is committed to creating a supportive, inclusive, and accessible environment for reputed company. If reasonable accommodation is needed to participate in the job application or interview process please let our recruiting team know. Apply tot his job Apply To this Job