Staff Machine Learning Engineer - Infrastructure jobs in United States
cer-icon
Apply on Employer Site
company-logo

Rad AI · 3 weeks ago

Staff Machine Learning Engineer - Infrastructure

Rad AI is a company on a mission to transform healthcare with artificial intelligence, focusing on revolutionizing radiology through AI-driven solutions. They are seeking a Staff Machine Learning Engineer to join their MLOps team to build and maintain the infrastructure that supports their AI research and products, ultimately improving clinical outcomes for patients.

Artificial Intelligence (AI)Enterprise SoftwareHealth Care
check
H1B Sponsor Likelynote

Responsibilities

Architect the infrastructure that supports our machine learning applications, services, and workflows
Architect and maintain our ML platform that supports continuous integration, continuous delivery, and continuous training for our machine learning models
Develop cloud-native services and serverless architectures to build scalable and resilient systems
Partner with data scientists to design the data pipeline that enable various machine learning models in production
Write code that meets our internal standards for security, style, maintainability, and best practices for a high-scale HIPAA web environment
Design, deploy, and maintain the full ML platform stack including monitoring and observability, data analytics, backend integration with customer-facing products, and the full model R&D lifecycle
Work with Product Management, Research, and Engineering to iterate on new features and address inefficiencies across our AI/ML infrastructure

Qualification

Machine Learning EngineeringPythonCloud ComputingInfrastructure ArchitectureKubernetesDockerDevOps ToolsMachine Learning FrameworksData Pipeline DesignCommunicationProblem-Solving

Required

8+ years of industry experience in ML Engineering in cloud-native environments
In-depth knowledge of Python (required)
Strong experience with infrastructure and DevOps tools such as Kubernetes, Docker, and Ansible
Strong knowledge of cloud computing platforms such as AWS (preferable), GCP, and Azure
Experience architecting distributed systems, storage systems, and databases
Experience working with machine learning frameworks such as PyTorch and LangGraph
Experience with Airflow (preferable) or other orchestration tools
Experience with infrastructure-as-code tools such as Terraform (preferable), Pulumi, Cloud Formation, etc
Experience with monitoring, tracing, and logging tools such Cloudwatch, NewRelic, Grafana, etc
Excellent communication skills, with a strong sense of ownership and a systematic approach to problem-solving
Proven ability to manage and lead active incidents, address what caused them, and establish systems to avoid them in the future via blameless postmortems

Preferred

Javascript/Typescript (nice to have)
Experience working with productionizing or optimizing inference of LLMs or other NLP models
Experience with the Ray ecosystem
Experience with PostgreSQL
Experience with data analytics tools like Hex, Amplitude, Retool, etc
Experience working at a fast-growing startup
Experience in a HIPAA-compliant environment

Benefits

Comprehensive Medical, Dental, Vision & Life insurance
HSA (with employer match), FSA, & DCFSA
401(k)
11 Paid Company Holidays
Location Flexibility (Remote-first company!)
Flexible PTO policy
Annual company-wide offsite
Periodic team offsites
Annual equipment stipend

Company

Rad AI

twittertwittertwitter
company-logo
Rad AI develops software that uses AI to automate radiology report writing, aiming to enhance accuracy and reduce radiologist burnout.

H1B Sponsorship

Rad AI has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (14)
2024 (7)
2023 (3)
2022 (6)
2021 (2)

Funding

Current Stage
Growth Stage
Total Funding
$151M
Key Investors
Transformation CapitalKhosla VenturesArtis Ventures
2025-05-20Series C· $8M
2025-01-25Series C· $60M
2024-05-07Series B· $50M

Leadership Team

leader-logo
Doktor Gurson
Co-founder & CEO
linkedin
leader-logo
Barrett Dwyer
CFO
linkedin
Company data provided by crunchbase