Job Description
Key Responsibilities Model Development & Optimization
· Independently design and train robust, reusable AI/ML models using frameworks like Kubeflow, PyTorch, TensorFlow, or HuggingFace.
· Apply advanced techniques in deep learning, NLP, computer vision, and classical ML, ensuring models are explainable and scalable.
· Optimize models for accuracy, speed, and resource efficiency, and collaborate with Data Engineers on deployment readiness. AI/ML System Design
· Translate business use cases into scalable AI model architectures, including design of pre-processing strategies and feature engineering approaches.
· Ensure alignment between data characteristics, model architecture, and expected system behavior in production.
· Collaborate on integration patterns with Data Engineers to embed AI models in downstream workflows. Documentation & Standards
· Write clear documentation on model assumptions, architecture choices, evaluation criteria, ...