TensorFlow Jobs in Sweden.
All active TensorFlow roles based in Sweden.
Pick a job to read the details
Tap any role on the left — its description and apply link will open here.
AI/ML Specialist Solutions Architect
Share this job
About Nebius:
Nebius is leading a new era in cloud infrastructure for the global AI economy. We are building a full-stack AI cloud platform that supports developers and enterprises from data and model training through to production deployment, without the cost and complexity of building large in-house AI/ML infrastructure.
Built by engineers, for engineers. From large-scale GPU orchestration to inference optimization, we own the hard problems across compute, storage, networking and applied AI.
Listed on Nasdaq (NBIS) and headquartered in Amsterdam, we have a global footprint with R&D hubs across Europe, the UK, North America and Israel. Our team of 1,500+ includes hundreds of engineers with deep expertise across hardware, software and AI R&D.
The role
We seek an experienced Specialist Solutions Architect to support AI-focused customers leveraging Nebius services. In this role, you will be a trusted advisor, collaborating with clients to design scalable AI solutions, resolve technical challenges and manage large-scale AI deployments involving hundreds to thousands of GPUs.
You’re welcome to work on-site in Amsterdam or remotely from any other EU country.
Your responsibilities will include:
- Designing customer-centric solutions that maximize business value and align with strategic goals.
- Building and maintaining long-term relationships to foster trust and ensure customer satisfaction.
- Delivering technical presentations, producing whitepapers, creating manuals and hosting webinars for audiences with varying technical expertise.
- Collaborating with engineering and product teams to effectively prioritize and relay customer feedback.
We expect you to have:
- 3+ years of experience with cloud technologies in MLOps engineering, Machine Learning engineering or similar roles.
- Strong understanding of ML ecosystems, including models, use cases and tooling.
- Proven experience in setting up and optimizing distributed training pipelines across multi-node and multi-GPU environments.
- Hands-on knowledge of frameworks like PyTorch or JAX.
- Excellent verbal and written communication skills.
It will be an added bonus if you have:
- Expertise in deploying inference infrastructure for production workloads.
- Ability to transition ML pipelines from POC to scalable production systems.
Preferred tooling:
- Programming Languages – Python, Go, Java, C++
- Orchestration – Kubernetes (K8s), Slurm
- DevOps Tools – Git, Docker, Helm
- Infrastructure as Code (IaC) – Terraform
- ML Frameworks and Libraries – PyTorch, TensorFlow, JAX, HuggingFace, Scikit-learn
Benefits & Perks:
- Competitive compensation
- Career growth and learning opportunities
- Flexibility and work-life balance
- Collaborative and innovative culture
- Opportunity to work on impactful AI projects
- International environment and talented teams
What's it like to work at Nebius:
Fast moving - Bold thinking - Constant growth - Meaningful impact - Trust and real ownership - Opportunity to shape the future of AI
Equal Opportunity Statement:
Nebius is an equal opportunity employer. We are committed to fostering an inclusive and diverse workplace and to providing equal employment opportunities in all aspects of employment. We do not discriminate on the basis of race, color, religion, sex (including pregnancy), national origin, ancestry, age, disability, genetic information, marital status, veteran status, sexual orientation, gender identity or expression, or any other characteristic protected by applicable law.
Applicants must be authorized to work in the country in which they apply and will be required to provide proof of employment eligibility as a condition of hire.
If you need accommodations during the application process, please let us know.
Ready to apply?
Apply to Nebius
Customer Engineer EMEA
Share this job
About Nebius:
Nebius is leading a new era in cloud infrastructure for the global AI economy. We are building a full-stack AI cloud platform that supports developers and enterprises from data and model training through to production deployment, without the cost and complexity of building large in-house AI/ML infrastructure.
Built by engineers, for engineers. From large-scale GPU orchestration to inference optimization, we own the hard problems across compute, storage, networking and applied AI.
Listed on Nasdaq (NBIS) and headquartered in Amsterdam, we have a global footprint with R&D hubs across Europe, the UK, North America and Israel. Our team of 1,500+ includes hundreds of engineers with deep expertise across hardware, software and AI R&D.
The role
We are looking for a Customer Engineer to support key and strategic Nebius GPU Cloud services customers. In this role, you will be a trusted technical advisor, helping clients design, deploy, and scale AI solutions while managing large-scale GPU workloads involving hundreds to thousands of GPUs. You will also collaborate with sales and product teams to drive growth and enhance customer satisfaction.
You’re welcome to work remotely from Europe.
Your responsibilities will include:
- Serve as the primary technical point of contact for troubleshooting and resolving complex AI/ML issues.
- Guide customers in optimizing GPU performance for ML training and inference workloads, ensuring seamless integration and scalability.
- Partner with the sales team to identify new opportunities, promote the latest products, and deliver technical presentations.
- Act as a bridge to product teams, providing customer feedback, relaying feature requests, and ensuring alignment with customer requirements.
- Engage with internal and external stakeholders, negotiate solutions, and effectively drive alignment to address customer challenges.
We expect you to have:
- Experience: 5+ years in roles like Solutions Architect, Technical Account Manager, or Customer Engineer, with hands-on experience in cloud services and AI/ML workloads.
- Proficiency in Infrastructure as Code (IaC) tools like Terraform and Ansible.
- Experience with Kubernetes and Python programming.
- Solid understanding of GPU computing, including ML training, inference workloads, and GPU stacks (e.g., CUDA, OpenCL).
- Customer-centric approach with a proven ability to build trust and foster long-term relationships.
- Strong ability to explain technical concepts to technical and non-technical audiences.
It will be an added bonus if you have:
- Hands-on experience with HPC/ML orchestration frameworks (e.g., Slurm, Kubeflow).
- Experience with deep learning frameworks (e.g., PyTorch, TensorFlow).
- Familiarity with ML tools from NVIDIA, AWS, Azure, and Google Cloud providers.
- Strong project management skills, with the ability to prioritize tasks and deliver on deadlines.
- Proven experience mentoring technical teams and driving team growth.
- Expertise in stakeholder negotiation to support problem resolution and ensure seamless collaboration.
Benefits & Perks:
- Competitive compensation
- Career growth and learning opportunities
- Flexibility and work-life balance
- Collaborative and innovative culture
- Opportunity to work on impactful AI projects
- International environment and talented teams
What's it like to work at Nebius:
Fast moving - Bold thinking - Constant growth - Meaningful impact - Trust and real ownership - Opportunity to shape the future of AI
Equal Opportunity Statement:
Nebius is an equal opportunity employer. We are committed to fostering an inclusive and diverse workplace and to providing equal employment opportunities in all aspects of employment. We do not discriminate on the basis of race, color, religion, sex (including pregnancy), national origin, ancestry, age, disability, genetic information, marital status, veteran status, sexual orientation, gender identity or expression, or any other characteristic protected by applicable law.
Applicants must be authorized to work in the country in which they apply and will be required to provide proof of employment eligibility as a condition of hire.
If you need accommodations during the application process, please let us know.
Ready to apply?
Apply to Nebius
Key Customers Solutions Architect EMEA
Share this job
About Nebius:
Nebius is leading a new era in cloud infrastructure for the global AI economy. We are building a full-stack AI cloud platform that supports developers and enterprises from data and model training through to production deployment, without the cost and complexity of building large in-house AI/ML infrastructure.
Built by engineers, for engineers. From large-scale GPU orchestration to inference optimization, we own the hard problems across compute, storage, networking and applied AI.
Listed on Nasdaq (NBIS) and headquartered in Amsterdam, we have a global footprint with R&D hubs across Europe, the UK, North America and Israel. Our team of 1,500+ includes hundreds of engineers with deep expertise across hardware, software and AI R&D.
The role
Nebius seeks a Key Customers Solutions Architect to support key and strategic Nebius GPU Cloud services customers. In this role, you will be a trusted technical advisor, helping clients design, deploy, and scale AI solutions while managing large-scale GPU workloads involving hundreds to thousands of GPUs. You will also collaborate with sales and product teams to drive growth and enhance customer satisfaction.
You’re welcome to work remotely from Europe.
Your responsibilities will include:
- Serve as the primary technical point of contact, troubleshooting and resolving complex AI/ML.
- Guide customers in optimizing GPU performance for ML training and inference workloads, ensuring seamless integration and scalability.
- Partner with the sales team to identify new opportunities, promote the latest products, and deliver technical presentations.
- Act as a bridge to product teams, providing customer feedback, relaying feature requests, and ensuring alignment with customer requirements.
- Engage with internal and external stakeholders, negotiate solutions, and effectively drive alignment to address customer challenges.
We expect you to have:
- Experience: 5 - 10 + years in roles like Cloud Solutions Architect, Technical Account Manager, or Customer Engineer, with hands-on experience in cloud services and AI/ML workloads.
- Proficiency in Infrastructure as Code (IaC) tools like Terraform and Ansible.
- Experience with Kubernetes and Python programming.
- Solid understanding of GPU computing, including ML training, inference workloads, and GPU stacks (e.g., CUDA, OpenCL).
- Customer-centric approach with a proven ability to build trust and foster long-term relationships.
- Strong ability to explain technical concepts to technical and non-technical audiences.
It will be an added bonus if you have:
- Hands-on experience with HPC/ML orchestration frameworks (e.g., Slurm, Kubeflow).
- Experience with deep learning frameworks (e.g., PyTorch, TensorFlow).
- Familiarity with ML tools from NVIDIA, AWS, Azure, and Google Cloud providers.
- Strong project management skills, with the ability to prioritize tasks and deliver on deadlines.
- Proven experience mentoring technical teams and driving team growth.
- Expertise in stakeholder negotiation to support problem resolution and ensure seamless collaboration.
Benefits & Perks:
- Competitive compensation
- Career growth and learning opportunities
- Flexibility and work-life balance
- Collaborative and innovative culture
- Opportunity to work on impactful AI projects
- International environment and talented teams
What's it like to work at Nebius:
Fast moving - Bold thinking - Constant growth - Meaningful impact - Trust and real ownership - Opportunity to shape the future of AI
Equal Opportunity Statement:
Nebius is an equal opportunity employer. We are committed to fostering an inclusive and diverse workplace and to providing equal employment opportunities in all aspects of employment. We do not discriminate on the basis of race, color, religion, sex (including pregnancy), national origin, ancestry, age, disability, genetic information, marital status, veteran status, sexual orientation, gender identity or expression, or any other characteristic protected by applicable law.
Applicants must be authorized to work in the country in which they apply and will be required to provide proof of employment eligibility as a condition of hire.
If you need accommodations during the application process, please let us know.
Ready to apply?
Apply to Nebius
Lead Cloud Solutions Architect
Share this job
About Nebius:
Nebius is leading a new era in cloud infrastructure for the global AI economy. We are building a full-stack AI cloud platform that supports developers and enterprises from data and model training through to production deployment, without the cost and complexity of building large in-house AI/ML infrastructure.
Built by engineers, for engineers. From large-scale GPU orchestration to inference optimization, we own the hard problems across compute, storage, networking and applied AI.
Listed on Nasdaq (NBIS) and headquartered in Amsterdam, we have a global footprint with R&D hubs across Europe, the UK, North America and Israel. Our team of 1,500+ includes hundreds of engineers with deep expertise across hardware, software and AI R&D.
The team
Customer experience at Nebius GPU AI Cloud involves tackling customers’ challenges and directly impacting their success by solving real-world AI and ML problems at massive GPU cloud scale. You’ll not only resolve issues, but play a key role in shaping clients’ business success by optimizing their AI solutions. Working with advanced GPUs such as H200, B200 and GB200, as well as modern ML frameworks, you’ll influence the development of the Nebius AI Cloud and gain experience at the intersection of infrastructure and AI. With minimal bureaucracy, you’ll have the freedom to innovate, take ownership and drive change. Opportunities for growth are abundant in this vibrant and supportive professional community.
The role
We are seeking a highly skilled and customer-focused leader to join our team as a Cloud Solutions Architect Lead. As a team leader, you will play a pivotal role in customer success across EU region, both managing the efforts of Cloud Solutions Architects team (technical and process excellence, talent development) and building processes alongside your stakeholders. Player-coach mentality and proven track record of managing high expertise teams is a must.
You’re welcome to work from our office in Amsterdam or remotely from any EU country.
Your responsibilities will include:
- Lead and mentor a team of Solutions Architects, fostering a culture of technical excellence, customer empathy, and collaborative execution.
- Oversee and participate in technical engagements including PoCs, workshops, architecture reviews, and trainings.
- Drive consistency and quality across infrastructure design, IaC practices, technical documentation, and customer enablement content.
- Collaborate closely with Product and Development teams, representing customer needs and helping shape the roadmap.
- Guide customers in optimizing ML/AI/HPC pipelines for performance, scalability, and cost-efficiency on Nebius cloud infrastructure.
- Build visibility and ownership culture amongst Solutions Architects.
We expect you to have:
- 5+ years of experience in a technical role (Solutions Architect, DevOps Engineer, ML Engineer, or similar) with a focus on cloud-native systems and distributed computing.
- 2+ years of experience leading or mentoring customer facing engineering teams.
- Deep hands-on experience with Infrastructure as Code (Terraform, Ansible), Kubernetes, and cloud automation.
- Strong communication and stakeholder management skills across both technical and non-technical audiences.
- Passion for developing scalable, reliable cloud architectures and driving customer success.
It will be an added bonus if you have:
- Experience with HPC or ML orchestration frameworks (e.g., Slurm, Kubeflow, Ray).
- Strong coding skills in Python and experience with GPU computing environments for ML/AI workloads.
- Proficient in GPU stack components (e.g., CUDA, cuDNN) and cloud-native patterns for ML training/inference.
- Familiarity with deep learning frameworks (e.g., TensorFlow, PyTorch, JAX).
- Understanding of cloud-based ML tools and offerings from NVIDIA, AWS, Azure, or Google Cloud.
- Experience participating in or leading technical contributions to conferences or open-source communities.
Benefits & Perks:
- Competitive compensation
- Career growth and learning opportunities
- Flexibility and work-life balance
- Collaborative and innovative culture
- Opportunity to work on impactful AI projects
- International environment and talented teams
What's it like to work at Nebius:
Fast moving - Bold thinking - Constant growth - Meaningful impact - Trust and real ownership - Opportunity to shape the future of AI
Equal Opportunity Statement:
Nebius is an equal opportunity employer. We are committed to fostering an inclusive and diverse workplace and to providing equal employment opportunities in all aspects of employment. We do not discriminate on the basis of race, color, religion, sex (including pregnancy), national origin, ancestry, age, disability, genetic information, marital status, veteran status, sexual orientation, gender identity or expression, or any other characteristic protected by applicable law.
Applicants must be authorized to work in the country in which they apply and will be required to provide proof of employment eligibility as a condition of hire.
If you need accommodations during the application process, please let us know.
Ready to apply?
Apply to Nebius