Machine Learning Engineer, Motion Planning & Prediction at Abridge

Austin, Texas, United States

Apply Now

Not SpecifiedCompensation

Mid-level (3 to 4 years), Senior (5 to 8 years)Experience Level

Full TimeJob Type

UnknownVisa

Autonomous Vehicles, RoboticsIndustries

Requirements

Strong proficiency in Python and hands-on experience with modern deep learning frameworks (e.g., PyTorch, TensorFlow, or JAX)
Solid understanding of machine learning fundamentals, including various neural network architectures, training methodologies, and evaluation techniques
Experience with the full machine learning lifecycle, from data exploration and prototyping to deployment and monitoring
Proficiency in C++ for writing high-performance model inference code
Authorized to work in the U.S. (no relocation sponsorship or remote work options available)

Responsibilities

Design, train, and deploy state-of-the-art machine learning models for behavioral prediction and motion planning
Develop robust data pipelines to process, clean, and label massive-scale vehicle sensor and simulation datasets
Work with deep learning architectures such as transformers to model complex temporal interactions between traffic agents
Establish and own the metrics for model performance, and create evaluation frameworks that correlate with on-road safety and performance
Collaborate with software engineers to integrate and optimize trained models for real-time inference on the vehicles embedded hardware
Stay current with the latest research in machine learning, imitation learning, and reinforcement learning, and apply novel techniques to our systems

Skills

Key technologies and capabilities for this role

PythonPyTorchTensorFlowJAXTransformersDeep LearningNeural NetworksData PipelinesMotion PlanningBehavioral PredictionImitation LearningReinforcement Learning

Questions & Answers

Common questions about this position

Is remote work available for this position?

No, remote work options are not available.

What are the required technical skills for this Machine Learning Engineer role?

The role requires strong proficiency in Python, hands-on experience with modern deep learning frameworks like PyTorch, TensorFlow, or JAX, solid understanding of machine learning fundamentals, experience with the full ML lifecycle, and proficiency in C++ for high-performance model inference.

What is the salary range for this position?

This information is not specified in the job description.

What is the team like at Abridge?

The team develops core software and data processing systems for motion planning and decision-making in autonomous vehicles, working at the intersection of machine learning, large-scale data infrastructure, and real-time vehicle control, while collaborating across engineering, analytics, and product teams.

What makes a strong candidate for this role?

Strong candidates will have the required skills in Python, deep learning frameworks, ML fundamentals, full lifecycle experience, and C++, plus nice-to-haves like a track record in ML competitions, robotics experience, MLOps tools, distributed data processing, or publications in top ML/robotics conferences.

Abridge

AI platform for clinical documentation automation

About Abridge

Abridge provides a platform that changes how medical conversations are documented in healthcare settings. Its main product allows healthcare providers to record patient visits, which are then transformed into organized clinical documents. This helps doctors and other healthcare professionals save time on paperwork, allowing them to concentrate more on caring for their patients. Abridge stands out from its competitors by focusing specifically on the needs of healthcare providers, payers, and pharmaceutical companies, offering a solution that improves both documentation efficiency and patient outcomes. The goal of Abridge is to enhance the quality of patient care while making the documentation process easier for healthcare professionals.

Pittsburgh, PennsylvaniaHeadquarters

2018Year Founded

$201.8MTotal Funding

SERIES_CCompany Stage

AI & Machine Learning, HealthcareIndustries

51-200Employees