AI Engineer
Eli Lilly and Company4 days ago
South San Francisco, CA, USAMid Level
Base Salary
$141k - $253k/yr
Responsibilities
- Develop and apply AI techniques to support clinical research and drug discovery.
- Build, train, evaluate, optimize, and deploy large-scale AI and machine learning systems.
- Perform distributed training on multi-GPU and multi-node infrastructure, including GPU performance profiling and optimization.
- Optimize and serve models for inference using production deployment and scheduling infrastructure.
- Build model evaluations and benchmarks that identify failure modes.
- Develop full-stack AI applications spanning data, models, backend services, APIs, and frontend interfaces.
- Design resilient, scalable, performant, and secure system architectures.
- Collaborate with AI Scientists, ML Ops Engineers, Data Engineers, and NVIDIA experts.
- Lead architecture reviews, code quality efforts, mentorship, and protection of proprietary data, models, and intellectual property.
- Follow ethical AI, data-handling, privacy, and regulatory requirements.
Requirements
- Master’s degree in Computer Science, Machine Learning, Artificial Intelligence, Data Science, Engineering, or a related technical field.
- At least 4 years of hands-on engineering experience building machine learning systems and moving models from research or prototypes into working systems.
- Advanced Python with production experience in PyTorch or JAX.
- Hands-on distributed training with multi-GPU and multi-node infrastructure, including DDP, FSDP, DeepSpeed, or Megatron.
- Experience with GPU performance profiling and optimization.
- Experience optimizing and serving models for inference with Triton, vLLM, or TensorRT-LLM.
- Experience with containerization and scheduling using Docker, Kubernetes, Ray, or Slurm.
- Experience building model evaluations and benchmarks.
- Backend development, frontend familiarity with JavaScript and a framework such as React, and API design experience.
- Familiarity with NumPy, SciPy, and Pandas.
- C++ or CUDA experience for performance-critical work is preferred.
- Experience with AWS, Azure, or GCP alongside on-premises GPU clusters.
- Strong problem-solving, communication, collaboration, and ambiguity-navigation skills.
- Experience with distributed training or large-scale model deployment on GPU infrastructure.
- Prior experience working across the full stack of an AI application.
Benefits
- Flexible hybrid work model with three days onsite and two days remote each week at the Silicon Valley Hub.
- Eligible full-time employees may receive a company bonus based partly on company and individual performance.
- Benefits may include a 401(k), pension, vacation, medical, dental, vision, prescription drug, flexible spending, life insurance, leave, well-being, fitness, employee assistance, and employee club benefits.
- Equal employment opportunity and workplace accommodation support are provided.
Tech Stack
Categories
About Eli Lilly and Company
Eli Lilly and Company researches, develops, manufactures, and markets prescription medicines and biologics for patients, sold through healthcare providers, pharmacies, and payers. Core therapeutic areas include diabetes, oncology, immunology, neuroscience, and obesity. Founded in 1876 and headquartered in Indianapolis, it is a public company (NYSE: LLY) with global operations and is building a new advanced manufacturing site for gene therapy and other modalities in Lebanon, Indiana.