ML Engineer for Production-Ready AI in Legal Tech

DeepJudge AG; Zürich; Hybrid; CHF 120,000 - 180,000; Full time

Job summary

DeepJudge AG is seeking a Machine Learning Engineer to design, build, evaluate and refine AI systems powering our core products. You will own the lifecycle of ML-enabled components from research to production, including data analysis, modeling, implementation and delivery to customers.

You will work on data analysis, model development and deployment, ensuring robust, scalable solutions that improve search and knowledge applications for legal teams.

Qualifications

• Experience training and deploying deep learning models.
• Proficiency with PyTorch, JAX or TensorFlow.
• Experience building scalable AI-powered systems.
• Familiarity with neural retrieval or agentic systems is a plus.

Responsibilities

• Bridge Research and Production: translate cutting‑edge research into scalable, production‑ready ML systems.
• Core ML: train, evaluate and optimize ML systems for robust, scalable deployment.
• Search: improve search quality with models, data and evaluation across offline/online modes.

Skills

Hands-on DL training

Model serving

PyTorch

TensorFlow

JAX

Tools

Git

Docker

Kubernetes

Job description

DeepJudge AG is seeking a Machine Learning Engineer to design, build, evaluate and refine AI systems powering our core products. You will own the lifecycle of ML-enabled components from research to production, including data analysis, modeling, implementation and delivery to customers.

You will work on data analysis, model development and deployment, ensuring robust, scalable solutions that improve search and knowledge applications for legal teams.