onsite
AI Software Engineer - Talent Vision Services
Software Engineer
AI Software Engineer focused on building production‑ready ML models using Python, TensorFlow, and PyTorch, deploying them on AWS with Docker orchestration. You’ll design scalable pipelines, collaborate with cross‑functional teams, and drive continuous improvement in model performance and reliability.
About the role
Key Responsibilities
- Design, develop, and deploy end‑to‑end machine learning pipelines in Python, leveraging TensorFlow and PyTorch frameworks.
- Implement model training, validation, and hyper‑parameter tuning workflows, ensuring reproducibility and scalability.
- Containerize ML services with Docker and orchestrate deployments on AWS (SageMaker, ECS, EKS) for high availability.
- Collaborate with data scientists, product managers, and DevOps to translate business requirements into robust AI solutions.
- Monitor model performance in production, troubleshoot issues, and iterate on model improvements.
Requirements
- 3+ years of software engineering experience with a strong focus on AI/ML.
- Proficiency in Python and experience with TensorFlow or PyTorch.
- Hands‑on experience deploying ML models on AWS using services such as SageMaker, ECS, or EKS.
- Solid understanding of containerization (Docker) and CI/CD pipelines.
- Excellent problem‑solving skills and a collaborative mindset.
Skills
pythonmachine learningtensorflowpytorchawsdocker