AI Engineer I

Ambry Genetics is seeking an entry-level AI Engineer to support the development, optimization, and deployment of machine learning models with a focus on GPU-based acceleration. The ideal candidate has strong programming skills, a foundation in deep learning, and hands-on experience with GPU infrastructure. You'll work closely with AI scientists, data engineers, and software developers to build scalable AI pipelines and tools for biomedical and genomic data analysis. Essential Functions: Develop, test, and deploy AI/ML models for genomics and biomedical applications Optimize training and inference performance using GPU-enabled frameworks (e.g., CUDA, PyTorch, TensorFlow) Maintain ML infrastructure including containerized services, APIs, and workflow automation Troubleshoot and improve existing AI pipelines and model performance Collaborate on software and system design to support cloud-based AI model serving Support CI/CD practices and version control for ML lifecycle tracking Contribute to model validation and benchmarking using open-source and proprietary datasets Other duties as assigned Qualifications: 1+ years of experience (including academic or internship) in AI/ML engineering or deep learning development BS or MS in Computer Science, Data Science, AI/ML, Bioinformatics, or related discipline or equivalent work experience Demonstrated experience training models on GPU infrastructure Programming experience in Python and frameworks like PyTorch, TensorFlow, scikit-learn Familiarity with containerization (e.g., Docker) and cloud infrastructure (AWS, GCP) Understanding of REST APIs, data serialization (JSON, Protobuf), and data management Version control using Git; knowledge of collaborative software development best practices Basic knowledge of relational databases (PostgreSQL, MySQL) and Linux environments Hands-on experience with NVIDIA CUDA, GPU optimization, or similar technologies Solid understanding of machine learning and deep learning principles Preferred: Knowledge of genomics or biomedical data is a plus

Ambry Genetics Corporation is a CAP-accredited and CLIA-licensed molecular genetics laboratory based in Aliso Viejo, California. We are a genetics-based healthcare company that is dedicated to open scientific exchange so we can work together to understand and treat all human disease faster. At Ambry, everyone is welcome. A career at Ambry Genetics is a chance to be part of a dynamic company that aims to improve health by understanding the relationships between genetics and human disease. We earned our reputation as industry leaders by responsibly introducing cutting-edge genetic testing solutions and continually sharing what we learn with the global scientific community. At Ambry you will be learning, challenging yourself, and having fun while collaborating with teammates through the open exchange of ideas. Our outstanding benefits program includes 401k, medical, dental, vision, FSA, paid sick leave and generous paid time off (PTO) program. Ambry Genetics is an Equal Opportunity Employer (EOE) and we maintain a drug-free work environment.