Founding Lead Machine Learning Engineer

3 days ago

Stockholm, Stockholm, Sweden A1 Full time

About A1

A1 is a self-funded, independent AI group, focused on building a new consumer AI application with global impact. We're assembling a small, elite team of ML, engineering and product builders who want to work on meaningful, high-impact problems.

About The Role

You will shape the core technical direction of A1 - model selection, training strategy, infrastructure, and long-term architecture. This is a founding technical role: your decisions will define our model stack, our data strategy, and our product capabilities for years ahead.

You won't just fine-tune models - you'll design systems: training pipelines, evaluation frameworks, inference stacks, and scalable deployment architectures. You will have full autonomy to experiment with frontier models (LLaMA, Mistral, Qwen, Claude-compatible architectures) and build new approaches where existing ones fall short.

What You'll Be Doing

  • Build end-to-end training pipelines: data → training → eval → inference
  • Design new model architectures or adapt open-source frontier models
  • Fine-tune models using state-of-the-art methods (LoRA/QLoRA, SFT, DPO, distillation)
  • Architect scalable inference systems using vLLM / TensorRT-LLM / DeepSpeed
  • Build data systems for high-quality synthetic and real-world training data
  • Develop alignment, safety, and guardrail strategies
  • Design evaluation frameworks across performance, robustness, safety, and bias
  • Own deployment: GPU optimization, latency reduction, scaling policies
  • Shape early product direction, experiment with new use cases, and build AI-powered experiences from zero
  • Explore frontier techniques: retrieval-augmented training, mixture-of-experts, distillation, multi-agent orchestration, multimodal models

What You'll Need

  • Strong background in deep learning and transformer architectures
  • Hands-on experience training or fine-tuning large models (LLMs or vision models)
  • Proficiency with PyTorch, JAX, or TensorFlow
  • Experience with distributed training frameworks (DeepSpeed, FSDP, Megatron, ZeRO, Ray)
  • Strong software engineering skills — writing robust, production-grade systems
  • Experience with GPU optimization: memory efficiency, quantization, mixed precision
  • Comfortable owning ambiguous, zero-to-one technical problems end-to-end

Nice to Have

  • Experience with LLM inference frameworks (vLLM, TensorRT-LLM, FasterTransformer)
  • Contributions to open-source ML libraries
  • Background in scientific computing, compilers, or GPU kernels
  • Experience with RLHF pipelines (PPO, DPO, ORPO)
  • Experience training or deploying multimodal or diffusion models
  • Experience in large-scale data processing (Apache Arrow, Spark, Ray)
  • Prior work in a research lab (Google Brain, DeepMind, FAIR, Anthropic, OpenAI)

  • Machine Learning Engineer

    2 weeks ago

    Stockholm, Stockholm, Sweden La Fosse Full time

    Job Title:Machine Learning EngineerLocation:Stockholm, SwedenContract:PermanentI am looking for a Machine Learning Engineer to join our client's team in Stockholm. In this role you'll take ownership of building and running production grade ML systems, focusing on real time predictions and the underlying platforms that support them.Your core...

  • Senior Machine Learning Engineer

    2 weeks ago

    Stockholm, Stockholm, Sweden Agaton Full time

    TL;DR - We want a data-obsessed ML engineer to unlock the intelligence within AI-powered customer conversations. You'll transform millions of customer interactions into predictive insights that power personalized coaching for consumer teams worldwide, turning complex behavioral data into game-changing intelligence that reshapes how businesses communicate...

  • Machine Learning Engineer

    5 days ago

    Stockholm, Stockholm, Sweden Coody Full time

    This is CoodyWe're one of Sweden's fastest-growing tech consultancies, expanding in both people and revenue. We attribute our growth to our simple yet powerful approach to Gravity-Free Consulting,combining the freedom and flexibility of freelancing with the security of traditional employment.At Coody, we believe in freedom, trust, and ambition which means...

  • Machine Learning DevOps

    3 days ago

    Stockholm, Stockholm, Sweden Pathway Full time

    About PathwayPathway is shaking the foundations of artificial intelligence by introducing the world's first post-transformer model that adapts and thinks just like humans. Pathway's breakthrough architecture (BDH) outperforms Transformer and provides the enterprise with full visibility into how the model works. Combining the foundational model with the...

  • Staff Machine Learning Engineer

    2 weeks ago

    Stockholm, Stockholm, Sweden Spotify Full time

    The Personalization team at Spotify makes deciding what to play next effortless and enjoyable for every listener. We aim to deeply understand music, podcasts, audiobooks, and videos to deliver exceptional recommendations that keep hundreds of millions of people engaged every day. Our work spans across experiences like Home, Search, curated playlists such as...

  • Senior Machine Learning Engineer

    2 weeks ago

    Stockholm, Stockholm, Sweden hormona Full time

    About UsHormona is a data-driven women's health company on a mission to revolutionize female health. Our award-winning app and world-first rapid hormone test form the only comprehensive, personalized platform that helps women understand and optimize their hormonal well-being. Hormona's innovative test delivers quantitative hormone results in just 15 minutes,...

  • Machine Learning Engineer

    2 weeks ago

    Stockholm, Stockholm, Sweden Market Partner Scandinavia Full time

    The world's most popular audio streaming subscription service is looking for aMachine Learning Engineerto join the band in a consultant assignment. The client has transformed music listening forever when launched in 2008.Period:ASAP to full-time), with a possibility of extension.About the roleFocus on these products:QUAIL. Building and applying ML models for...

  • Machine Learning Engineer

    2 weeks ago

    Stockholm, Stockholm, Sweden Professional Galaxy AB Full time

    Professional Galaxy is looking for a Machine Learning Engineer (focused on QUAIL + MAP integration) on behalf of our clientFocus on these products:QUAIL. Building and applying ML models for language evaluation, quality scoring, and multimodal translation workflows.MAP (Multilingual AI Platform). It is a platform to translate content at scale using AI. MAP is...

  • Machine Learning Engineer

    5 days ago

    Stockholm, Stockholm, Sweden GS1 Sweden Full time

    Machine Learning Engineer till GS1 SwedenVill du utveckla, implementera och optimera ML-modeller i produktion? Som Machine Learning Engineer hos GS1 Sweden får du en nyckelroll i att omsätta avancerade AI-koncept till robusta, skalbara lösningar.Om tjänstenVisste du att du möter oss varje dag, ofta utan att veta om det? Vi är organisationen bakom...

  • Founding Engineer

    7 days ago

    Stockholm, Stockholm, Sweden Few&Far Full time

    Founding Full-Stack Engineer - Stockholm - OPEN TO RELOCATIONI'm currently working with a well-funded AI startup that's quietly building one of the most ambitious products in the agentic systems space, and they're looking for strongFounding Full Stack Engineersto join the teamThe company has recently closed a$6M seed round led by a top-tier global VCand is...