
Tech Data - AI/ML Engineer, Enterprise AI Suite
- Pune, Maharashtra
- Permanent
- Full-time
- direct interaction with product owners and internal users to identify requirements, development of technical solutions and execution
- develop an SDK (Software Development Kit) to automatically capture Data Product, Dataset and AI / ML model metadata. Also, leverage LLMs to generate descriptive information about assets
- integration and publication of metadata into UBS's AI Use-case inventory, model artifact registry and
- design and implementation of services that seamlessly collects runtime evidence and operational information about a data product or model and publishes it to appropriate visualization tools
- creation of a collection of starters/templates that accelerate the creation of new data products by leveraging a collection of the latest tools and services and providing diverse and rich experiences to the Devpod ecosystem.
- design and implementation of data contract and fine-grained access mechanisms to enable data consumption on a 'need to know' basis
- phd or Master's degree in Computer Science or any related advanced quantitative discipline
- 5+ years industry experience with Python / Pandas, SQL / Spark, Azure fundamentals / Kubernetes and Gitlab
- additional experience in data engineering frameworks (Databricks / Kedro / Flyte), ML frameworks (MLFlow / DVC) and Agentic Frameworks (Langchain, Langgraph, CrewAI) is a plus
- ability to produce secure and clean code that is stable, scalable, operational, and well-performing. Be up to date with the latest IT standards (security, best practices). Understanding the security principles in the banking systems is a plus
- ability to work independently, manage individual project priorities, deadlines and deliverables
- Willingness to quickly learn and adopt various technologies
- excellent English language written and verbal communication skills