Principal Machine Learning Engineer

2 months ago

H1B Sponsor

Base Salary

$275k - $350k/yr

Responsibilities

Interpret qualitative challenges in building AI agents for science as optimizable problems.
Build environments for training and deploying AI agents for scientific tasks.
Collaborate with scientists to formulate and scale training data pipelines.
Lead training of large-scale LLM-based systems and improve experimentation infrastructure.
Build efficient inference infrastructure supporting complex sampling algorithms.
Develop and extend the experimentation platform for internal tools and projects.
Foster collaboration with a multidisciplinary team of AI researchers and scientists.

8-10+ years of experience in applied ML research and real-world problem solving.
Experience across the ML lifecycle including data pipelines, model training, and deployment.
Fluency in PyTorch, Jax, or equivalent frameworks.
Demonstrated experience with experimentation in academic or industry settings.
Strong programming expertise to tackle various technical challenges in ML and LLM software.

Competitive salary and equity.
Full healthcare coverage for you and your dependents.
Support for growing families, including a yearly new parent stipend and fertility coverage.
401(k) company matching.
$300 health and wellness benefit.
Daily lunch and dinner provided when working late.
Regular team offsites and company events.
A fast-moving, mission-driven culture.

PyTorch