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.