Software Engineer - Machine Learning

Every day, the world gets more digital thanks to tens of millions of developers building the future faster than ever. But with exponential growth comes exponential risk, as outnumbered security teams struggle to secure mountains of code. This is where Snyk (pronounced “sneak”) comes in. Snyk is a developer security platform that makes it easy for development teams to find, prioritize, and fix security vulnerabilities in code, dependencies, containers, and cloud infrastructure — and do it all right from the start. Snyk is on a mission to make the world a more secure place by empowering developers to develop fast and stay secure.

Joining Snyk means embracing our core values: One Team, Care Deeply, Customer Centric, and Forward Thinking. As a member of our team, you’ll have the opportunity to thrive in a dynamic environment where fostering collaboration, leading with empathy, driving business impact, and inspiring trust are at the heart of everything we do.

Overview

We are seeking a highly skilled and motivated Machine Learning Engineer with a strong desire to improve the code analysis and remediation, to join the Machine Intelligence Team. We own the process from ideation to product, in a strong collaboration with other groups and divisions. Example of a feature we own is, but not limited to, is Snyk DeepCode AI Fix.

The ideal candidate will have industrial-level (research and/or production) ML experience of 3+ years, and a proven track record of developing production-grade ML pipelines. This role requires a deep understanding of language model optimizations and the ability to run and own experimentation end-to-end. The successful candidate will be proficient in C++, Python, one of typical ML training and serving frameworks, and will play a crucial role in advancing our machine learning capabilities for code analysis and program understanding.

You’ll spend your time:

• Design, implement, and maintain machine learning models at the intersection of program analysis and language processing, such as automatic code generation, code understanding.
• Develop and optimize production-grade pipelines to ensure scalability and efficiency.
• Optimize existing machine learning models for performance and efficiency.
• Lead cross-team collaborations to integrate machine learning solutions into our products and services.
• This is a "research engineer" position – we expect that you own features end-to-end, from ideation at our regular small-hands hackathons, to production.

You should apply if you:

• Fluency in programming languages, specifically C++ and Python.
• ML-serving savviness: you saw something as Triton/TensorRT/ONNX/llama.cpp in action.
• Demonstrated experience in creating and maintaining production-grade data-, training- and evaluation pipelines.
• Expertise in model optimization and the ability to manage experimentation processes end-to-end.
• Familiarity with build systems such as Bazel (or similar), CI/CD pipelines, ability to operate at high testing standards.

We’d especially love to hear from you if you:

• Ability to not stop at the demo-level: Jupyter notebook, or even a locally running Python script is not your final goal.
• Familiarity with ML training, evaluation, and experiment tracking frameworks.
• Additional backend-heavy languages such as Go, Rust.
• Experience with production/serving stack, such as Docker, k8s.
• Research activities, publications in the field of ML, contributions to open-source projects.
• Prior experience in the security domain, code analysis or program understanding.
• Experience with large-scale machine learning projects.

#LI-JW1

We care deeply about the warm, inclusive environment we’ve created and we value diversity – we welcome applications from those typically underrepresented in tech. If you like the sound of this role but are not totally sure whether you’re the right person, do apply anyway!

About Snyk

Snyk is committed to creating an inclusive and engaging environment where our employees can thrive as we rally behind our common mission to make the digital world a safer place. From Snyk employee resource groups, to global benefits that help our employees prioritize their health, wellness, financial security, and a work/life blend, we aim to support our employees along their entire journeys here at Snyk.

Benefits & Programs

Prioritize health, wellness, financial security, and life balance with programs tailored to your location and role.

• Flexible working hours, work-from home allowances, in-office perks, and time off for learning and self development
• Generous vacation and wellness time off, country-specific holidays, and 100% paid parental leave for all caregivers
• Health benefits, employee assistance plans, and annual wellness allowance
• Country-specific life insurance, disability benefits, and retirement/pension programs, plus mobile phone and education allowances