Note: The job is a remote job and is open to candidates in USA. ImmunityBio, Inc. is a biotechnology company focused on developing innovative therapies to enhance the immune system's ability to combat cancer. The Software Engineer – AI will play a crucial role in integrating data engineering and AI systems, collaborating with cross-functional teams to develop and maintain production-grade AI solutions.
Responsibilities
- Contribute to the design, development, maintenance, and deployment of AI agent systems, including one or more of LangGraph StateGraph patterns, CrewAI multi-agent orchestration, LlamaIndex data/agent workflows, or related frameworks
- Design, build, and maintain FastAPI-based agent servers, including async/await endpoints, streaming responses, and health check endpoints, following established architecture and coding standards
- Implement comprehensive testing strategies — unit, integration, edge-case, and performance — for agents and services to ensure production reliability
- Implement and extend guardrails for AI interactions, including input/output validation, safety checks, prompt hardening, policy enforcement, and robust error handling in alignment with organizational security and compliance standards
- Implement secure tool execution patterns adhering to least-privilege access, allowlists, secrets management, audit logging, and validation controls against prompt injection/data exfiltration under guidance from senior engineers
- Implement agent observability including per-step traces, tool-call telemetry, cost/latency budgets, and SLO-based alerting using existing monitoring and logging platforms
- Help design and maintain agent evaluation sets, adversarial tests, regression tests, and monitoring for safety/quality drift
- Design and build RAG/retrieval pipelines using vector databases such as Chroma, Milvus, Weaviate, and Qdrant, working with data and platform teams on schema and performance
- Collaborate with SMEs across Platform Engineering, Data Science, Security, Quality, and Product to ensure agent systems meet technical and operational requirements
- Contribute to the creation, modification, and maintenance of AI system documentation, including architecture decision records (ADRs), runbooks, API specifications, network topology diagrams, data flow diagrams, and best-practice guides
- Provide technical input regarding operability, technical feasibility, engineering design, security posture, maintainability, and documentation requirements
- Assist with the planning and implementation of AI agent-based projects and solutions in collaboration with cross-functional teams, helping deliver on cost, timeline, and quality targets
- Implement and support containerized deployment, start-up, commissioning, and release qualification activities using Docker multi-stage builds, dependency management, and container health check best practices
- Collaborate with and learn from senior engineers; may occasionally assist less experienced engineers through code reviews and pairing on troubleshooting AI agent and service-related issues
- Follow and improve Standard Operating Procedures (SOPs), process improvements, and standard engineering templates for agent development
- Monitor agent performance and reliability, collect metrics and logs, and use data to propose and implement optimization opportunities
- Performs ad-hoc and cross-functional duties and/or projects as assigned to support business needs and provide developmental opportunities
Skills
- Bachelor's degree in Computer science, Software Engineering, or a related field with 3+ years of relevant software development experience is required
- 2+ years of hands-on Python development experience, with proficiency in Python 3.12+ and modern language features (type hints, async/await, etc.) is required
- 2+ years of hands-on Typescript development experience
- Experience with at least one AI agent framework (LangGraph, LangChain, CrewAI, or LlamaIndex) in a production or near-production setting is required
- Experience building and running containerized applications with Docker, including multi-stage builds and foundational DevOps practices, is required
- Experience with troubleshooting and system optimization in an enterprise or regulated environment is preferred
- Experience working within an organization with formal change control, compliance, or audit requirements is preferred
Benefits
- Medical, Dental and Vision Plan Options
- Health and Financial Wellness Programs
- Employer Assistance Program (EAP)
- Company Paid and Voluntary Life/AD&D, Short-Term and Long-Term Disability
- Healthcare and Dependent Care Flexible Spending Accounts
- 401(k) Retirement Plan with Company Match
- 529 Education Savings Program
- Voluntary Legal Services, Identity Theft Protection, Pet Insurance and Employee Discounts, Rewards and Perks
- Paid Time Off (PTO) includes: 11 Holidays
- Exempt Employees are eligible for Unlimited PTO
- Non-Exempt Employees are eligible for 10 Vacation Days, 56 Hours of Health Pay, 2 Personal Days and 1 Cultural Day
Company Overview
Company H1B Sponsorship