Software Engineer - Hosted Model Infrastructure
As a Software Engineer on the Hosted Model Infrastructure team, you will build and operate high-performance model serving infrastructure, deployment pipelines, and observability for production AI systems. This role involves working across the full stack to enable ML models in various environments, including air-gapped government networks and edge nodes.
About the job
The Role
We are a software engineering team with expertise in enabling ML models in production. We deploy AI models to run in variety of environments: air-gapped government networks, forward-deployed defense environments, edge nodes, and enterprises with strict data sovereignty requirements. Our customers rely on us for frontier AI capabilities running on hardware they control, often with constrained GPU resources and limited direct access. Rising to that challenge and meeting those expectations is what Palantir's excels at.We treat models like any other software: continuously tested, continually delivered, packaged for reproducible deployment, and built for long-term maintainability. You will own services end-to-end, and work across the full stack, from inference engines, GPU scheduling to deployment pipelines, observability, and integration with Palantir's platform. The goal is to deliver new models and capabilities quickly and continuously.
Join us if you want to solve problems at the intersection of infrastructure and machine learning that directly enable critical customers.
Technologies We Use
- Different backend languages, including Java, Rust, Python and Go
- Model serving engines for GPU-accelerated inference
- Docker and Kubernetes for containerization and orchestration
- Industry-standard build tooling, including Gradle and GitHub
Core Responsibilities
- Building high-performance model serving infrastructure that integrates with security models, hardware constraints, and different inference engines
- Designing intelligent request handling including authentication, rate limiting, concurrency control, and audit logging for multi-tenant model access
- Building and maintaining packaging and deployment pipelines enabling fast, secure, and reliable model rollouts across on-premises and air-gapped environments
- Developing observability for production AI systems to enable easy service monitoring and fast incident triage and resolution
- Debugging complex issues and performance problems throughout the stack, including open source inference engines, container runtimes, and GPU drivers, in environments you cannot always access directly
- Designing and running testing and benchmarking infrastructure that validates model deployments across varying GPU hardware before they reach production
- Working with product teams and customers to understand requirements, debug production issues, and deliver the models and capabilities they need
- Integrating hosted model infrastructure with Palantir's deployment, configuration, and identity systems
What We Value
- Ownership mindset and bias toward quality. Our software runs in environments where direct access for debugging is limited or unavailable.
- High empathy for customer needs and drive to deliver reliable, easy-to-use models
- Ability to work effectively across multiple languages and layers of the stack, from backend services and ML tooling to container orchestration and deployment configuration
- Strong debugging skills and motivation to trace problems from application code through containers, orchestration, and hardware
- Curiosity about emerging AI capabilities and the ability to quickly evaluate and integrate new models and technologies as the landscape evolves
- Active US Security clearance, or eligibility and willingness to obtain a US Security clearance is beneficial, but not necessary
What We Require
- 4+ years of professional software engineering experience building and operating production systems
- Engineering background in Computer Science, Mathematics, Software Engineering, Physics, or similar field
- Strong coding skills with demonstrated proficiency in programming languages, such as Java, C++, Python, Rust, or similar languages. Familiarity with the Python ML ecosystem is valuable.
- Experience with containers, Kubernetes, and deploying backend services in production environments
- Strong written and verbal communication skills and ability to iterate quickly with teammates, incorporating feedback and holding a high bar for quality
Skills
Java, Rust, Python, Go, Docker, Kubernetes, Gradle, GitHub, GPU, Machine Learning
Similar jobs
ML Engineering jobs
Thinking Machines Lab
San Francisco, CA
Build and operate the engineering systems that support post-training research, including reinforcement learning infrastructure, sandboxed execution, data pipelines, and agent scaffolding. The role requires strong Python and systems engineering skills, project ownership, and a relevant bachelor’s degree or equivalent experience.
OpenAI
San Francisco, CA
Build research infrastructure and tooling that enables AI models to design silicon, including reinforcement learning environments, EDA integrations, evaluations, and experiment workflows. The role requires strong software engineering fundamentals and comfort working across research, tooling, and chip-design systems.
Rollstack
United StatesBuild production AI capabilities for automated slide and document generation, working across LLM applications, data analysis, and content generation. The role requires 3+ years in machine learning and NLP, advanced Python, and experience with LLM frameworks and production systems.
ClickUp
United States
Build and operate large-scale ranking and retrieval systems that power search relevance, including hybrid lexical/vector search, embeddings, query understanding, and permission-aware retrieval. Requires a bachelor's degree and 5+ years of ML engineering experience in ranking or information retrieval.
PathAI
Boston, MADevelop and deploy machine learning models for biomedical research and AI products, collaborating with scientific, engineering, and product teams. Requires an advanced quantitative degree, substantial ML experience, Python proficiency, and experience bringing models into production or research applications.