Staff AI Engineer

Type: Full-time

Location: Lausanne

About us

At Visium, we enable enterprise executives in defining their AI & Data strategy, execute large scale transformations and implement AI across operations, ensuring their organization becomes future-proof.

With expertise in strategy, architecture, cloud engineering, analytics, artificial intelligence and machine learning, we empower our clients to unleash and scale the power of their data.

We’re on a mission to pioneer a bright future and build future-proof and ethical organizations . Join the curious, the ambitious, the doers, the good-hearted, the ones who build a world we’re all in awe of – our Visiumees.

Ready to become one?

Role

As a Staff Machine Learning Engineer,  you will own the most technically complex and high-impact projects from conceptualization over architecture design to production deployment. 

You will:

• Set direction for a technical team, ensuring architectural excellence and long-term solution maintainability.
• Lead and mentor junior and senior engineers, raising the technical bar across projects.
• Stay hands-on with high quality code contributions to data processing, modeling and deployment pipelines while delegating effectively.
• Partner with clients to translate complex business challenges into clear technical roadmaps and deliver measurable business outcomes.
• Drive adoption of modern ML best practices and state-of-the-art methodology from system design and architecture over modelling to MLOps.

You are not only passionate about machine learning, but also about shaping how data-driven insights create value for our clients. You thrive in dynamic environments, balancing hands-on technical work with strategic decision-making and team leadership.

This role goes beyond building models: you will be responsible for technical strategy, quality standards and growing engineering talent, shaping the future of our ML delivery practice.

To succeed in this role, you bring a combination of expertise, experience, and skills including:

• 6 - 8 years of experience in machine learning, data science, or related fields.
• Proficiency in Python programming.
• Expertise in ML/DL libraries and frameworks (e.g., Scikit-learn, TensorFlow, Keras, PyTorch).
• Experience with data visualization tools (e.g., Matplotlib, Seaborn, Power BI, Tableau, D3.js).
• Deep learning experience (e.g., RNNs, CNNs) and strong knowledge of clustering, regression, and classification methods.
• Experience in designing, developing, and deploying LLM-based applications, including RAG systems, with a strong understanding of prompt engineering, vector databases, and knowledge retrieval pipelines.
• Familiarity with Unix/Linux environments, REST APIs, and microservices.
• Experience with ML deployment, production pipelines, and scalable architectures on platforms such as AzureML, AWS and GCP. 
• Experience with Data and AI platforms (Databricks, Dataiku, Snowflake). Certifications are a plus. 
• Demonstrated technical and project leadership, with the ability to guide complex initiatives and deliver results.
• Strong mentorship skills and experience managing or coaching teams to reach their full potential.
• Excellent collaboration and communication skills, capable of bridging technical and business perspectives.
• Fluent in English, French is a plus. 

What we offer

• A yearly education budget to steep your learning curve
• A yearly sport budget because a fit body leads to a fit mind
• A position that enables you to have an impact on 1’000s of people
• An international, knowledgeable, and passionate team with a strong collaborative mindset
• Opportunity to join a talented and experienced company with proven traction in its journey
• An open and transparent culture

Check our LinkedIn and Instagram to learn more about us & don’t hesitate to contact us if you have any questions.