Skip to content
PalantirPalantirNew York, NY

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.

Salary not listed
Hybrid4+ YOEML Engineering

About the role

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

JavaRustPythonGoDockerKubernetesGradleGitHubGPUMachine Learning

Similar roles

ML Engineering jobs
Cerebras Systems

CoDesign & NextGen Performance Engineer

Cerebras SystemsSunnyvale, CA

Characterize, analyze, and optimize performance of state-of-the-art AI models on Cerebras' wafer-scale hardware. Build performance models, optimize kernels and compilers, debug runtime behavior, and develop visualization tools to influence next-gen AI architecture.

Salary not listed
On-site3+ YOEML Engineering
OpenAI

Research Engineer, Privacy

OpenAISan Francisco, CA

Research Engineer on OpenAI's Privacy team designing and prototyping privacy-preserving ML algorithms like differential privacy and federated learning at scale. Requires hands-on PETs experience, fluency in PyTorch/JAX, and a track record implementing or publishing novel privacy work.

380k – 445k/yr
HybridML Engineering
Console

Research Engineer

ConsoleSan Francisco, CA

Research Engineer building self-improving AI agent systems at Console. Develop eval/optimization loops, fine-tune specialist models, and improve agent reasoning over enterprise context using production data to drive measurable gains in quality, latency, and reliability.

200k – 350k/yr
On-siteML Engineering
Notion

Software Engineer, AI Platform

NotionSan Francisco, CA +1

Build and scale the shared AI platform foundations at Notion, enabling fast and safe shipping of AI products. Requires experience with LLM/ML platforms, strong ownership, and comfort across backend, infrastructure, and product code.

180k – 201k/yr
Hybrid5+ YOEML Engineering
Liftoff

Machine Learning Engineer

LiftoffCalifornia

Machine Learning Engineer building statistical models, optimization systems, and experiments for mobile ad tech economics on the Revenue Engine team. Requires PhD in CS/ML/Economics and industry experience applying ML or economics at scale.

215k – 275k/yr
RemoteML Engineering