Machine Learning Engineer

TalentDome Staffing is partnering with a rapidly growing, U.S.-based AI startup that is set to revolutionize compliance workflows in the financial services industry.

Our client is building a neutral compliance layer to manage complex, document-heavy processes at scale. They are a pre-Series A company with a small, highly technical team of 17 (including 3 in the US). They are looking for a hands-on, mid-level Machine Learning Engineer to join their core team, work alongside the Senior Data Scientist and Machine Learning Engineer based in the U.S.

This is a 100% "hands-on" role. We are looking for a "doer" who is passionate about building, training, and deploying models from the ground up.

What You’ll Do

As a core Machine Learning Engineer, you will be responsible for the entire model lifecycle. You won't just be running analytics; you'll be building the engine.

• Design, build, train, and deploy machine learning models using PyTorch (preferred) and TensorFlow.
• Own the end-to-end model development process: from data gathering and feature engineering (handling outliers, normalization) to training and implementation.
• Perform rigorous model evaluation and validation, and be able to clearly explain your choice of metrics (e.g., F1-score, ROC/AUC, Accuracy vs. Precision).
• Work with a variety of model architectures (Transformers, CNNs, RNNs, XGBoost, etc.) to find the best solution for the problem.
• Collaborate closely with the senior data science and architecture team to rapidly prototype, iterate, and ship models to production.

What You'll Bring (The "Must-Haves")

We are looking for a mid-level engineer with a "startup attitude." The right mindset is more important than years of experience.

• Experience: 3-5 years of hands-on experience as a Machine Learning Engineer.
• Core Skill: A strong, practical understanding of PyTorch (highly preferred) and/or TensorFlow.
• ML Fundamentals: Deep, practical knowledge of ML fundamentals. You must be able to confidently explain the difference between accuracy and precision, what an F1-score is, and how to read a ROC curve.
• Proven "Builder": You must have experience building and training models "from the ground up," not just running analytics in a notebook or using pre-built APIs.
• Feature Engineering: Strong skills in feature engineering, including handling outliers, normalization, and cleaning skewed data.
• Startup Attitude (Essential): You are a "doer" who is passionate, driven, and independent. You thrive in a fast-paced environment, are willing to wear multiple hats, and have an "ask for forgiveness, not permission" mentality.

What We Offer

• A competitive base salary + bonus
• Equity in a high-growth company.
• A comprehensive benefits package.
• A 100% fully remote work environment.

Please note: You must be a resident of Sweden and have unrestricted work authorization to be eligible for this role.