ML Infrastructure Engineer jobs in United States
cer-icon
Apply on Employer Site
company-logo

VantAI · 22 hours ago

ML Infrastructure Engineer

Proxima is a company that combines advanced machine learning techniques with systems biology expertise to develop innovative computational models for drug discovery. The ML Infrastructure Engineer will be responsible for developing large-scale scientific workflows, automating processes, and supporting scientists in enhancing existing workflows to drive efficiency in drug discovery efforts.

Alternative MedicineArtificial Intelligence (AI)BiotechnologyClinical TrialsLife ScienceMachine Learning
check
H1B Sponsor Likelynote

Responsibilities

Shaping our product by spearheading new features and services
Developing software tools to automate or improve processes ranging from data ingestion pipelines to bespoke cheminformatics workflows
Contributing to the culture of a rapidly growing company
Being challenged by your colleagues and learning something new every day

Qualification

PythonKubernetesML infrastructureData architectureParallel computingCloud-native architecturesCheminformaticsBioinformaticsLarge language modelsIndependenceProblem-solving

Required

Experience working in complex codebases and cloud-native architectures
Container orchestration and Kubernetes cluster development
Demonstrated experience creating and managing infrastructure for ML model registry, training, inference and deployment infrastructure
Strong familiarity with best practices and current state of the art for ML infrastructure
Experience supporting large-scale model architectures with complex environment requirements
Data architecture and engineering
Demonstrated experience with parallel computing and distributed model architectures
Ability to break apart existing code; solve ad hoc problems with little help
Willingness to work with existing codebases
No task too big or small; strong sense of independence and ownership
Provide guidance and opinions on implementation/considerations, while having a willingness to compromise

Preferred

Developing/supporting ML models in the cheminformatics, bioinformatics, large language, and/or diffusion model domains (desired)
Experience working on drug discovery projects (desired)

Company

VantAI

twittertwittertwitter
company-logo
VantAI pairs bleeding-edge machine learning techniques with deep systems biology expertise to build computational models that uncover hidden relationships between molecules, targets, and diseases.

H1B Sponsorship

VantAI 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 (1)
2020 (1)

Funding

Current Stage
Early Stage
Total Funding
$80M
Key Investors
DCVC
2026-01-13Seed· $80M

Leadership Team

leader-logo
Zachary Carpenter
Founder & CEO
linkedin
leader-logo
Luca Naef
Co-Founder & CTO
linkedin
Company data provided by crunchbase