Senior AI Engineer
Amgen · Thousand Oaks, California
Protein Language Models & Multi-Omics Infrastructure
- Build ML infrastructure and evaluation workflows for protein-language-model and multi-omics use cases within Amgen's generative-biology environment.
- Benchmark AMPLIFY's 120M- and 350M-parameter checkpoints on protein sequences up to its published 2,048-residue context.
- Track task metrics, model configurations, sequence-length test slices, and efficiency regressions in MLflow to make checkpoint comparisons reproducible.
- Analyze task-quality and compute trade-offs between the 120M and 350M checkpoints by sequence-length slice; the published AMPLIFY 350M comparison reports 43× fewer parameters and 24–29× higher inference throughput than ESM2-15B, depending on sequence length.
- Engineer Spark SQL and Delta Lake pipelines for multi-omics data managed through AWS HealthOmics and Databricks.
- Package selected model services for controlled deployment with Amazon Bedrock, Docker, and Kubernetes.
Core technologies: Python, Databricks, Spark SQL, Delta Lake, MLflow, AWS HealthOmics, Bedrock, S3, Docker, Kubernetes