Senior Machine Learning Engineer

Location: Stockholm, Sweden (100% onsite).

Schedule: 1 FTE (40 hours per week)

Contract: Contract by June 30th, 2026, with a high possibility of extension.

Salary: Up to 38,96 EUR/h (non-negotiable) 

We are Baltic Assist, an outsourcing company, specializing in innovative remote career solutions. We provide exciting opportunities on a global level fostering long-term cooperation between top talent and our partners. By cultivating a desirable environment, our people are empowered to excel and engage with a community of diverse professionals. With a commitment to efficiency and innovation, Baltic Assist aims to be the go-to choice for seeking rewarding careers.

What you'll be doing

We’re looking for a Senior Machine Learning Engineer to join our team and help build, maintain, and optimize scalable machine learning and data pipelines in a cloud environment.

You’ll take ownership of designing robust solutions, implementing best practices in coding and testing, and collaborating with both technical teams and stakeholders to ensure high-quality, maintainable outcomes.

How you’ll spend your day

• Organizing, managing, and governing data assets to ensure consistency, reliability, and discoverability.
• Implementing and maintaining cloud infrastructure using structured, repeatable processes.
• Building, optimizing, and maintaining data pipelines and workflows for efficiency and scalability.
• Writing maintainable, high-quality code following software engineering best practices and structured design principles.
• Developing and maintaining automated tests to ensure system reliability and prevent regressions.
• Managing code changes collaboratively, reviewing contributions, and maintaining smooth deployment workflows.
• Containerizing and deploying applications, ensuring consistent and stable environments.
• Designing and maintaining data models, performing transformations, and contributing to overall system architecture.
• Collaborating effectively with stakeholders and cross-functional teams, clarifying requirements, providing guidance, and ensuring alignment.
• Participating in code reviews, mentoring peers, and maintaining team-wide quality standards.
• Proactively tackling challenges, troubleshooting complex problems, and applying structured approaches to new or unfamiliar scenarios.

You'll be successful if you have

• Familiarity with Data Catalog
• Understanding of Infrastructure as Code
• Cloud-agnostic skills
• Deep knowledge of Python programming, practicing OOP, following coding best practices, and knowing how to use flake8, mypy, black, SonarQube, and pre-commit
• Deep knowledge in unit tests and end-to-end tests, familiar with Pytest, fixtures, unittest, etc.
• Familiarity with Unix systems, knowing how to install applications in Docker, and using shell scripting
• Ability to manage code with Git, create pull requests, resolve merge conflicts, and implement CI/CD pipelines using GitHub Actions or similar
• Deep understanding of Docker
• Deep knowledge of DBT, preferably in GCP
• Deep knowledge of SQL
• Strong understanding of data modeling and system design
• Can-do attitude and proactive problem-solving, even with unfamiliar challenges
• Strong communication skills with stakeholders and technical teams
• Passion for code review and providing constructive feedback

Nice to have

• Knowledge of Dataflow
• Experience with Kubernetes (K8s)

What we offer

• Flexible working arrangements to help you succeed in your career while balancing personal needs.
• An inspiring environment in a large organisation.
• Work in an international team, which concentrates on innovative business-facing solutions.
• An extensive training program to ensure that your skills are maintained and updated on an ongoing basis.
• Open and friendly company culture and great colleagues.

At Baltic Assist, we believe that the strength of our team lies in the diversity of our backgrounds and perspectives. We are fully committed to promoting equal opportunities at every stage of our recruitment and selection processes. Recruitment at Baltic Assist is based solely on individual merit, qualifications, and professional competence. Our shortlisting, interviewing, and selection practices are conducted without discrimination on the grounds of gender identity or expression, sexual orientation, marital or civil partnership status, race, nationality, ethnic origin, religion or beliefs, age, veteran status, disability, medical condition, pregnancy or parental status, trade union membership, or any other protected characteristics.

By applying to this position, candidates acknowledge and consent that any information provided during the application process or available on the communication channels through which they were contacted (e.g., LinkedIn, email, or other professional platforms) may be collected, stored, and processed by Baltic Assist solely for recruitment purposes.

Baltic Assist is committed to protecting your personal data in compliance with the General Data Protection Regulation (GDPR). All information collected during the recruitment process will be handled securely and used exclusively for recruitment purposes. Your data will not be shared with unauthorized parties, and it will be retained only as long as necessary for the recruitment process or as required by applicable laws. By applying, you consent to the processing of your personal data in accordance with our privacy policy and GDPR guidelines.