Senior Machine Learning Engineer (m/f/d) in Berlin | momox SE

Your mission

• Design and implement robust machine learning solutions for real-world applications at scale
• Take algorithms and models from research to production, ensuring reliability, scalability, and observability
• Build and maintain ML pipelines using modern MLOps tools (e.g., Kubeflow Pipelines, MLflow, Weights & Biases, Evidently AI)
• Deploy and serve models using frameworks like FastAPI , Docker and Kubernetes , ensuring efficient retraining and monitoring
• Collaborate cross-functionally with data scientists, engineers, and product managers to translate business needs into ML-driven solutions
• Contribute to the evolution of momox’s ML system, and advocate for best practices in model lifecycle management and continuous improvement

Our tech stack

• Python: PyTorch, FastAPI, Scikit-learn, HuggingFace, LitServe, vLLM
• Cloud & Infrastructure: Docker, Kubernetes, GCP
• CI/CD: Gitlab, ArgoCD
• Model orchestration: Kubeflow (Vertex AI pipelines)
• Data: BigQuery, Postgres, Milvus, FAISS, FEAST
• Monitoring: Grafana, Evidently AI, W&B, Prometheus

Your profile

• Master’s or PhD in Computer Science , Engineering , Mathematics , Statistics or a related field focused on Machine Learning
• 3-5+ years of experience developing and deploying ML systems in production environments
• Deep understanding of ML algorithms, statistical modeling, and deep learning techniques
• Proficiency with PyTorch and at least one other modern ML framework such as JAX or TensorFlow
• Solid experience with MLOps , model orchestration, and observability tools (e.g., Prometheus , CI/CD , MLflow )
• Experience working with cloud platforms ( GCP , AWS or Azure )
• Strong communication skills - you can clearly explain complex ML concepts to both technical and non-technical audiences
• Fluency in English (German is a plus but not required)

Nice to have:

• Hands-on experience with vision-language models, reinforcement learning or recommender systems
• Knowledge of distributed training and large-scale inference
• Familiarity with Model Context Protocol (MCP) and emerging agentic workflows
• Understanding of safe GenAI application design and evaluation
• Experience using vector databases (e.g., Milvus, Pinecone) for retrieval-augmented systems
• Domain experience in pricing, logistics or marketing ML applications

You can look forward to

• Flexible working hours : We work according to a hybrid model that offers you maximum flexibility while also fostering important personal connections. Whether in our centrally located office or working remotely - you and your team decide together where and when you want to work. You can arrange your working hours flexibly. For personal interaction, your team meets at least 20 days per quarter in the office, as our team days are held in person.
• Training: momox is only as good as you. That's why you will find room and budget for your individual professional development with language classes, internal training, conferences, (online) courses and more.
• Company discounts : As a momoxian, you can look forward to a monthly voucher for our momox Fashion online shop, as well as medimops. Additionally, you benefit from numerous company discounts from our partners.
• Individual work environment : Design your ergonomic workspace just the way you like it – with height-adjustable desks and the laptop of your choice (MacBook or ThinkPad). If things get a bit noisy in the office, your Bose noise-cancelling headphones ensure uninterrupted focus. If needed, we can also provide you with an ergonomic chair or additional monitors for your home office.
• 1/2 day off on your birthday : Celebrate your birthday with half a day of vacation! Come in later, leave earlier, or simply take an additional half day off to have your entire birthday free – provided your birthday falls on a regular working day. Additionally, you get half a day off on December 24th and 31st.
• Office dogs : Our office dogs contribute to a relaxed work atmosphere and are always happy to receive some extra cuddles.
• Sustainability : We love second-hand and incorporate our sustainable business idea into the office routine with simple methods like using eco-friendly supplies, double-sided printing, energy-saving modes on devices, waste separation, etc. In doing so, we take responsibility and integrate sustainability into all our activities.
• Team events : Having fun at work and celebrating our successes together are core to our values. That’s why we regularly come together at various events to connect, get to know each other better, and strengthen our team spirit.
• Sport and fitness offers : During the week, you have the opportunity to take part in various sports activities. Whether it's online training, massages, or stretching at your desk – there's something for everyone. You can also get a discounted membership for Urban Sports Club or Wellhub.
• Reduced ticket for the public transport : Save real money with the Germany or Job Ticket for the Berlin S-Bahn.