Staff Machine Learning Engineer, Public Sector
Scale AI
27 days ago
New York, NY, USA +3 more
Staff+
H1B Sponsor
Base Salary
$260k - $327k/yr
Responsibilities
- Lead the architecture and implementation of agentic AI systems.
- Build and scale agents for complex geospatial reasoning.
- Design and improve retrieval systems for static and semi-structured documents.
- Fine-tune and evaluate embedding models for mission-critical datasets.
- Design memory systems for agents to persist state and learn from interactions.
- Own and evolve shared agentic infrastructure and core libraries.
- Define evaluation strategies for agentic systems, including robustness testing.
- Collaborate with engineering managers, product leaders, and researchers.
- Serve as a technical mentor to enhance system design and ML rigor.
Requirements
- 8+ years of experience building and deploying applied ML systems.
- Deep experience with agentic systems and autonomous workflows.
- Strong background in ML systems engineering, including model serving.
- Hands-on experience with retrieval systems and representation learning.
- Proficiency in Python and modern ML frameworks like PyTorch.
- Ability to operate at Staff-level scope and drive initiatives to production.
- Experience making tradeoffs across performance, cost, and reliability.
Benefits
- Comprehensive health, dental, and vision coverage.
- Retirement benefits and a learning and development stipend.
- Generous PTO and potential commuter stipend.
Tech Stack
PythonPyTorch
Categories
AI & MLData Science