Senior Software Engineer, Machine Learning Accelerators

4 weeks ago


San Francisco, California, United States Aurora Innovation Full time

About the Role

Aurora Innovation is seeking a talented Software Engineer to join our team and focus on ML Accelerators. In this role, you will work closely with our autonomy and hardware teams to understand our on-vehicle ML technology.

Key Responsibilities

  • Develop perspectives on where opportunities and gaps might be in our machine learning software and hardware strategy with a longer-term horizon in mind
  • Maintain relationships and intel across the ML ecosystem including both established players and start-ups (accelerators, software modules, etc.) for both existing and future products
  • Execute internal research and development of ML software and hardware technology
  • Surface high-impact findings to relevant Engineering leadership, keeping feedback loop going to influence Aurora's ML strategy
  • Help drive value to key engineering stakeholders

Requirements

  • Minimum 5+ years of professional experience
  • BS, MS, or PhD in Computer Science or related field
  • Strong programming skills in C++, Python
  • Familiarity with at least one deep learning framework (PyTorch, TensorFlow, MXNet, etc.)
  • Basic understanding of computer vision deep learning models
  • Strong analytical skills, especially for performance troubleshooting (e.g. profiling, roofline model)
  • Familiarity with CUDA, OpenCL, OpenVX, Halide or other SIMD programming models
  • Familiarity with ML-specific accelerators (NPUs, TPUs, IPUs, etc...)
  • Ability to quickly learn and adapt to new technology
  • Ability to work on large code bases and a fast growing environment
  • Strong communication skills
  • Comfortable with Linux/other unix environments

Preferred Qualifications

  • Experience with inference on edge platforms
  • Experience with cloud ML training pipelines
  • HPC experience

What We Offer

Aurora Innovation offers competitive medical, dental, and vision benefits, and additional healthcare support including medical transportation reimbursement, fertility, adoption, and surrogacy benefits. We empower our employees and their families with options to further their unique physical, mental, and financial well-being.

How to Apply

If you are a motivated and talented Software Engineer looking to join a dynamic team, please submit your application.



  • San Francisco, California, United States Aurora Innovation Full time

    About the RoleAurora Innovation is seeking a talented Software Engineer to focus on ML Accelerators. As a key member of our team, you will work closely with our autonomy and hardware teams to understand our on-vehicle ML technology.Key ResponsibilitiesDevelop perspectives on where opportunities and gaps might be in our machine learning software and hardware...


  • South San Francisco, California, United States Vaxart Full time

    We are seeking a Senior Software Engineer with expertise in machine learning to join our team at {company}.This role will involve:Designing and developing complex software applications using machine learning algorithmsCollaborating with cross-functional teams to integrate machine learning models into our productsStaying up-to-date with the latest...


  • San Francisco, California, United States Relyance AI Full time

    Job SummaryRelyance AI is seeking a highly skilled Senior Software Engineer, Machine Learning to join our team. As a key member of our engineering team, you will be responsible for designing, developing, and deploying machine learning models and algorithms to drive business outcomes.Key ResponsibilitiesDevelop and implement machine learning models and...


  • San Francisco, California, United States Plaid Full time

    We're looking for a skilled software engineer to join our Payment Risk team at Plaid. As a software engineer on this team, you will work on tooling and infrastructure that facilitates the development of our machine learning based risk models, from data and feature pipelines to model evaluation framework.You will also work on model serving infrastructure to...


  • San Francisco, California, United States WEX Full time

    About the RoleWe are seeking a highly motivated Senior Machine Learning Engineer to join our AI Engineering team at WEX. As a key member of our team, you will design, implement, train, and maintain machine learning algorithms and ML system pipelines. You will work closely with our AI platform team and line of business engineering teams to integrate AI...


  • San Bernardino, California, United States Department of Public Health Full time

    Job Description:We are seeking a Senior Software Engineer with proficiency in machine learning to lead our AI and data science projects. The ideal candidate will have a strong background in computer science and experience in developing and deploying machine learning models. The successful candidate will be responsible for designing and implementing AI...


  • San Francisco, California, United States Discord Full time

    About the RoleWe are seeking a highly skilled Senior Software Engineer to join our Machine Learning Platform team at Discord. As a key member of our team, you will be responsible for designing and building the platform that ML engineers and data scientists use to understand and delight Discord's users and keep them safe.Key ResponsibilitiesDesign and build...


  • San Francisco, California, United States Sephora Full time

    Job Summary: We are seeking a highly skilled Senior Machine Learning Engineer to join our team at Sephora. As a key member of our technology group, you will be responsible for driving the operationalization of AI/ML initiatives across the enterprise. This is an exciting opportunity to work on a broad set of domains that power a data-driven transformation of...


  • San Francisco, California, United States Tailored Management Full time

    Job Title: Senior Machine Learning EngineerLocation: Remote (PST/CST Preferred)Contract: 12 months with possible extensionPay Rate: $105/hour on W2 along with medical + Dental + Vision, PTO, Sick time.Job Description:Our client is embarking on a transformative change to its business and technology, and our Machine Learning Engineers are at the forefront of...


  • San Francisco, California, United States Tailored Management Full time

    About the Role:We're seeking a highly skilled Senior Machine Learning Engineer to join our team at Tailored Management. As a key member of our engineering team, you'll play a critical role in driving the development of innovative machine learning solutions that transform our business.Key Responsibilities:Develop and implement scalable machine learning models...


  • San Francisco, California, United States Unreal Gigs Full time

    Company Overview:Welcome to Unreal Gigs, a pioneer in AI-driven innovation. We're committed to building robust infrastructure that powers our machine learning models at scale. Our team is shaping the future of AI infrastructure engineering.Position Overview:As a Senior Machine Learning Infrastructure Architect, you'll lead the design, development, and...


  • San Jose, California, United States Zscaler Full time

    About ZscalerZscaler is a leading cloud security company that provides a comprehensive security platform to protect enterprises from cyber threats. As a Senior Software Engineer, Machine Learning, you will play a key role in developing and maintaining our machine learning-powered security solutions.Key Responsibilities:Design and develop large-scale...


  • San Jose, California, United States Tik Tok Full time

    Job DescriptionTikTok is a leading destination for short-form mobile video, and our mission is to inspire creativity and bring joy. Our platform is built to help imaginations thrive, and we're looking for a talented Senior Machine Learning Engineer to join our SMB engineering team.The SMB engineering team is responsible for developing software to empower...


  • San Francisco, California, United States EPAM Full time

    Role OverviewEPAM is seeking a highly skilled Machine Learning Engineer to contribute to the development and maintenance of on-device ML applications.Key Responsibilities:Develop and optimize data models and pipelines using Python and relevant data processing frameworks.Engage in data analysis to derive insights and make data-driven decisions.Implement basic...


  • San Francisco, California, United States Unreal Gigs Full time

    Unlock the Power of AIAt Unreal Gigs, we're on a mission to harness the potential of machine learning to drive innovation and growth. As a Machine Learning Engineer, you'll play a critical role in designing, implementing, and optimizing intelligent systems that unlock new opportunities for our clients.Key Responsibilities:Design and Develop Machine Learning...


  • San Francisco, California, United States Faire Inc Full time

    About FaireFaire is an innovative online wholesale marketplace that empowers independent retailers to thrive in a rapidly changing retail landscape. Our mission is to level the playing field for small businesses by leveraging the power of technology and data.At Faire, we're passionate about using machine learning to drive business growth and create a more...


  • San Francisco, California, United States Selby Jennings Full time

    About the RoleSelby Jennings is partnering with a leading FinTech company to find a seasoned Machine Learning Engineer. This innovative startup has disrupted the financial industry with its cutting-edge technology, making capital more accessible to everyone.The TeamThe team is comprised of experienced professionals from finance and technology backgrounds....


  • San Francisco, California, United States Unreal Gigs Full time

    Unreal Gigs OverviewWelcome to Unreal Gigs, a pioneering force in AI-driven innovation. We're committed to building robust infrastructure that powers our machine learning models at scale.Salary: $195,000 - $255,000 per yearPosition SummaryWe're seeking a seasoned Senior Machine Learning Infrastructure Engineer to lead the design, development, and...


  • San Francisco, California, United States Databricks Full time

    Pioneering a cutting-edge Generative AI platform, Databricks is seeking a highly skilled Senior Staff Machine Learning Engineer to spearhead the development of our ML infrastructure and drive our technology vision & roadmap.Key Responsibilities:• Oversee projects and engineers, ensuring seamless collaboration and technical excellence.• Design and...


  • San Francisco, California, United States Discord Full time

    Machine Learning Platform Software EngineerDiscord is a leading platform for gamers and communities, with over 200 million active users each month. Our platform plays a unique role in the future of gaming, and we're looking for a skilled software engineer to join our Machine Learning Platform team.This position is responsible for designing and building the...