STAGE Machine Learning Engineer – Monitoring et Intégration LLM en production

Key Skills

Large Language Models - LLMs, Natural Language Processing (NLP), Python Programming, PyTorch, TensorFlow, Transformer

Job Description

Stellantis has a transversal Data Science entity which supports all Machine Learning subjects, in particular NLP & Time Series subjects. We are looking to strengthen our team by welcoming a talented intern as a Machine Learning Engineer to help us monitor our LLM algorithms and integrate them into an MLflow infrastructure on the Databricks platform. If the trainee progresses well, we will be able to analyze how to generally deploy these models (transformer type) on Edge devices (IoT). The real-time aspect (in terms of deployment and retraining) can be a challenge in the Time Series part. Main Mission:

• Collaborate with our research and development team to continuously monitor the performance of our LLM models.
• Develop monitoring and reporting tools to identify potential degradations in model performance. Propose and implement solutions to improve the performance and stability of LLM models.
• Work on integrating these models into an MLflow based API with Databricks.
• Propose deployment methods in the Edge Responsibilities:
• Analyze and understand the monitoring needs of LLM models.
• Design relevant metrics and performance indicators.
• Develop scripts and automated tools for real-time monitoring.
• Collaborate with the development team for the integration of these models into MLflow and Databricks.
• Document monitoring and integration processes. What we offer:
• An exciting opportunity to work with a cutting-edge team in the field of AI.
• Access to cutting-edge resources and data to carry out your mission.
• The opportunity to learn and grow in a dynamic environment.
• A friendly and collaborative work environment. If you are passionate about AI, machine learning and the challenges of monitoring and integrating LLM models, we would love to have you on our team.

• Our benefits reflects the STELLANTIS commitment to helping you reach your personal and professional goals. In addition to an environment that promotes career development, we offer benefits for a healthy lifestyle and a rewarding future, designed to take care of you and your family, in various stages of life.

• As a global company, our employee packages will vary by country, customary norms and the legal entity into which you are hired.

• Student in computer science, data science, machine learning or related field.
• Knowledge of NLP
• Proficiency in machine learning tools and libraries (Python, TensorFlow, PyTorch, Transformers, etc.).
• Experience with MLflow and Databricks would be an asset.
• Strong programming and software development skills.
• Good communication and teamwork skills.
• Ability to work independently and solve problems creatively.

• Our benefits reflects the STELLANTIS commitment to helping you reach your personal and professional goals. In addition to an environment that promotes career development, we offer benefits for a healthy lifestyle and a rewarding future, designed to take care of you and your family, in various stages of life.

• As a global company, our employee packages will vary by country, customary norms and the legal entity into which you are hired.