Staff Machine Learning Engineer

Company Description

At The Stepstone Group, we help everyone get the job that best fits their life. Whether it is the next step in their career, or a job to pay the bills - we exist to match lifestyles with livelihoods.  

We play our part by giving people everything they need to find the right job for them in that moment; making jobs work for more people, whatever they do, and however they choose to do it. 

Equal opportunities are important to us. We believe that diversity and inclusion at Totaljobs is critical to our success as a global company, so we want to recruit, develop, and keep the best talent on the market. We encourage applications from everyone, regardless of your background, gender identity, sexual orientation, disability status, ethnicity, belief, age, family or parental status, and any other characteristic. If you require reasonable accommodation in completing this application or interviewing, please let us know. 

We also have a hybrid approach! Work from home 2 days and 3 days at the office!

Job Description

You will work closely together with 2 teams of data scientists & machine learning engineers. You will help us to improve the design of AI systems (classical ML & LLMs) on AWS and to draft and build new solutions. Such solutions include data enrichment pipelines and building blocks to train, evaluate, and serve large language models.  

What the role will look like on a day to day basis:

• Work together with the AI labs, Data, Infrastructure, and Platform teams to design and deploy ML-based services using AWS Sagemaker and other cloud based solutions 
• Analyse current AWS account setup for data scientists & MLEs and improve it together with StSt MLE & DevOps experts  
• Understand current AI pipelines and their integration with other services 
• Design & build infrastructure to facilitate experimentation & deployment of AI systems 
• Collaborate with the AI Labs teams to advance current ML workflows and software development standards  

Qualifications

• Industry experience in AWS ML services, including AWS Sagemaker 
• Experience in DevOps, Cloud Operations, ML-engineering 
• Experience in AWS administration and permission management 
• Expertise with building enterprise scale ML & LLM solutions 
• High proficiency with Docker, IaC (Terraform), CI/CD (Atlassian Bamboo) 
• Ambition to understand, question, & shape our ML-based services 
• Provide guidance on IaC and CI/CD best practices to increase the efficiency in deployment and management of resources.  
• Good communication skills in English, both written and verbal.  
• Experience with building and or using model management tools (e.g. mlflow) 
• Experience with MLOps (model orchestration, updating, etc.) 
• Experience with Kafka is a strong plus as some of the pipelines are event-Driven 
• Nice to have: Experience with setting up & managing Vector Databases. 
• Nice to have: Experience in deploying LLMs at scale.  

Additional Information

Benefits and Rewards:

We believe in giving recognition to our employees, so if you are interested to find out more, please look at the The Stepstone Group career site.

• Holiday - We feel work-life balance is important, so we offer every single one of our employees 29 days holiday a year, plus public holidays.  Everyone also has the option to buy or sell up to three days each year in our Holiday Flex scheme. 
• Family Leave - We understand that laying the foundations for a smooth start to family leave will also make for an easier transition back to the workplace and subject to qualifying criteria, the benefits under the Company Family schemes are more generous than required under current legislation.
• Pension Scheme - To help our people plan for their future we offer an excellent pension scheme.
• Medical - Access to digital GP and well-being services.
• Charity - Each employee receives x2 additional days off every year to volunteer to charitable causes. We also give our employees the chance to 'give as they earn' meaning they can donate tax-free to a registered charity, and The Stepstone Group will match their contribution up to a value of £10 per month.

When you apply for a role with us you will be asked to complete an equal opportunities questionnaire. If you have any questions about this, please feel free to reach out to one of the team.