What You'll Be DoingAs an AI Engineer, you will turn advanced AI models into reliable tools that support drug discovery. You will build, train, evaluate, and deploy AI systems, working closely with AI Scientists to shape model design and technical decisions. This is a hands-on role with opportunities to contribute to both engineering and scientific innovation. You'll collaborate with AI Scientists, ML Ops Engineers, Data Engineers, and NVIDIA experts to develop cutting-edge AI solutions.
How You'll Succeed- AI and Automation Execution: Develop and apply innovative AI techniques to solve complex business challenges, creating tailored solutions that drive strategic value and transform clinical research processes that support AI scientists.
- Full Stack Development: Build and implement AI-driven models and applications across the full stack, from backend services to frontend interfaces.
- System Architecture and Design: Design and uphold resilient system architectures to guarantee optimal performance, scalability, and security across all platforms.
- Handle large scale model development.
- Cross-Disciplinary and Organization Collaboration: Effectively engage with colleagues to gather innovative ideas and insights, fostering a collaborative environment that inspires and motivates team members to innovate and explore creative solutions in AI development.
- Ethics and Compliance in Development: Adhere to ethical guidelines in AI usage and data handling, ensuring compliance with all relevant regulations and maintaining the highest standards of data privacy and security.
- Elevate engineering excellence through architecture reviews, code quality leadership, and mentorship, while safeguarding Lilly's proprietary data, models, and intellectual property.
What You Should Bring- Advanced Python with production experience in PyTorch or JAX, and a track record of taking machine learning models from research code to working systems.
- Hands-on distributed training on multi-GPU, multi-node infrastructure (DDP, FSDP, DeepSpeed, or Megatron), with GPU performance profiling and optimization.
- Experience optimizing and serving models for inference (Triton, vLLM, or TensorRT-LLM) with containerization and scheduling (Docker, Kubernetes, Ray, or Slurm).
- Experience building evaluation and benchmarking for models - including designing the tests that reveal where a model fails, not only where it succeeds.
- Full stack capability - a backend language, front-end familiarity (JavaScript with a framework such as React), and API design - enough to build the interfaces through which scientists actually use a model.
- Comfort with the scientific Python stack (NumPy, SciPy, Pandas); C++ or CUDA for performance-critical work is a strong plus.
- Experience with cloud AI/ML platforms (AWS, Azure, or GCP) alongside on-premises GPU clusters.
- Excellent problem-solving skills and the ability to navigate ambiguity in a highly technical environment.
- Strong written and verbal communication and demonstrated ability to partner well with research scientists.
- Experience with distributed training or large-scale model deployment on GPU infrastructure.
- Prior experience working across the full stack of an AI application - data, model, service, and interface.
Your Basic Qualifications- Master's degree in Computer Science, Machine Learning, Artificial Intelligence, Data Science, Engineering, or a related technical field
- 4+ years of hands-on engineering experience building machine learning systems, including taking models from research or prototype into real working use.
Location & Work FlexibilityThis role is based at our Silicon Valley Hub. We offer a flexible hybrid work model, with
three days onsite and two days working remotely each week, supporting both collaboration and work-life balance.
Actual compensation will depend on a candidate's education, experience, skills, and geographic location. The anticipated wage for this position is
$141,000 - $253,000
Full-time equivalent employees also will be eligible for a company bonus (depending, in part, on company and individual performance). In addition, Lilly offers a comprehensive benefit program to eligible employees, including eligibility to participate in a company-sponsored 401(k); pension; vacation benefits; eligibility for medical, dental, vision and prescription drug benefits; flexible benefits (e.g., healthcare and/or dependent day care flexible spending accounts); life insurance and death benefits; certain time off and leave of absence benefits; and well-being benefits (e.g., employee assistance program, fitness benefits, and employee clubs and activities).Lilly reserves the right to amend, modify, or terminate its compensation and benefit programs in its sole discretion and Lilly's compensation practices and guidelines will apply regarding the details of any promotion or transfer of Lilly employees.
#WeAreLilly