Job description
Join Aurora AI Labs, a leading innovator in enterprise AI solutions. We are seeking a Senior AI Engineer to design, develop, and deploy cutting-edge ML models and AI tooling that power production systems for global customers.
In this role, you will partner with data scientists, software engineers, and platform teams to build scalable ML pipelines, apply state-of-the-art algorithms, and ensure reliability, governance, and interpretability of models in production.
Responsibility
- Lead design and implementation of production-grade ML models and AI systems.
- Collaborate with data scientists to translate research into scalable, maintainable production code.
- Architect robust data pipelines, feature stores, and data quality monitoring.
- Develop and maintain ML Ops tooling for training, validation, deployment, and monitoring.
- Optimize models for latency, throughput, cost; implement quantization and distillation as needed.
- Integrate ML services with cloud infrastructure (AWS/GCP/Azure) and CI/CD pipelines.
- Mentor engineers, participate in code reviews, and promote best practices.
- Drive governance, safety, and explainability considerations for deployed models.
Qualification
- Master's or PhD in Computer Science, Electrical Engineering, Statistics, or a related field; equivalent practical experience considered.
- 5+ years of professional experience in AI/ML engineering with production systems.
- Strong Python proficiency and hands-on experience with PyTorch or TensorFlow; experience with JAX is a plus.
- Experience with ML Ops tooling (MLflow, Kubeflow, Airflow) and cloud platforms (AWS, GCP, Azure).
- Solid understanding of data structures, algorithms, statistics, and model evaluation.
- Experience deploying and monitoring models in production, including A/B testing and performance optimization.
- Familiarity with data governance, security, and compliance considerations.
- Excellent collaboration, communication, and problem-solving skills; ability to operate in a fast-paced environment.