Machine Learning Research Engineer

Celonis is the global leader in Process Intelligence and the pioneer of Process Mining technology. As one of the world’s fastest-growing enterprise SaaS companies, we are changemakers pushing the boundaries of what’s possible. We invest heavily in advanced AI capabilities—specifically our Process Intelligence Graph—to turn data insights into immediate business action. We believe there is a massive opportunity to unlock global productivity and sustainability by placing intelligence at the core of every business process. Join our mission to make processes work for people, companies, and the planet. The Role As a Machine Learning Research Engineer, you will work at the intersection of AI research and software engineering. You will develop systems that analyze and optimize enterprise operations. This role requires collaboration across product and engineering teams to translate machine learning models into scalable product features. What you will do Design, test, and productionize machine learning algorithms for enterprise applications. Research, develop and iterate on foundation model architectures for specialized domains. Build and maintain ML infrastructure, pipelines, and backend services to support model deployment (MLOps). Integrate Generative AI capabilities (such as LLMs and RAG architectures) into the core Celonis platform to drive automation and process insights. Process and analyze complex operational datasets and knowledge graphs to train predictive models. Implement engineering best practices, ensuring code is efficient and testable, and utilizing logging, monitoring, and model-drift alerting. Collaborate with product managers, data engineers, and frontend developers to deploy end-to-end AI solutions. What you will need Education: A Master's or Ph.D. in Computer Science, Artificial Intelligence, Machine Learning, Mathematics, or a highly related field. Core Experience: 3+ years of experience in academia or industry, specializing in machine learning, deep learning, NLP, or artificial intelligence. ML/AI Expertise: Practical experience with Generative AI technologies, including Large Language Models (LLMs), fine-tuning, Prompt Engineering, and Retrieval-Augmented Generation (RAG). Additionally, comfort with established Machine Learning methods and paradigms, e.g. bias-variance tradeoff, overfitting, across a variety of settings (supervised, unsupervised, semi-supervised, structured, RL) are a necessary prerequisite. Engineering Foundation: Production-level experience in Python (and frameworks like PyTorch, TensorFlow, or JAX), alongside exposure to at least one other programming language (e.g., Java, Go, or C++). Cloud & DevOps: Experience deploying models using Docker, Kubernetes, and cloud compute services (AWS EC2/Lambda/SageMaker, Azure, or GCP). Data Handling: Proficiency in writing complex SQL queries, working with graph databases, and processing large datasets. Skills: Strong analytical and problem-solving abilities, with excellent English communication skills (German is a plus). Visa sponsorship is not offered for this role. The base salary range below is for the role in the specified location, based on a Full Time Schedule. Total compensation package will include base salary + bonus/commission + equity + benefits (health, dental, life, 401k, and paid time off). Please note that the base salary range is a guideline, and that the actual total compensation offer will be determined based on various factors, including, but not limited to, applicant's qualifications, skills, experiences, and location. The base salary range below is for the role in New York, based on a Full Time Schedule. $161,000 — $189,000 USD What Celonis can offer you: Pioneer Innovation: Work with the global leader in Process Mining and the Process Intelligence Graph to shape the future of AI-driven business operations. Ownership from Day 1: Every full-time "Celonaut" is an owner, receiving Restricted Stock Units (RSUs) and merit-based refresh grants. Unrivaled Family Support: Benefit from our inclusive parental leave policy—24 weeks of fully paid leave for primary carers and 12 weeks for supporting carers, available from your first day of employment. Work-Life Integration: Enjoy Unlimited PTO (in applicable regions) and generous PTO globally, as well as a flexible hybrid work model that balances remote focus with vibrant office collaboration. Continuous Growth: Elevate your skills through our 70-20-10 learning framework, mentorship programs, and access to a dedicated learning platform. Holistic Well-being: Prioritize your health with subsidized Wellhub memberships, mental health counseling, and dedicated "Wellness Weeks" that prioritize work/life balance. Drive Sustainability: Participate in annual Impact Days, where you receive paid time off to volunteer for community and environmental causes with your local office, or virtually. Global Inclusion & Belonging: Find co