Machine Learning Engineer

About Protect AI

Protect AI is shaping, defining, and innovating a new category within cybersecurity around the risk and security of AI/ML. Our ML Security Platform enables customers to see, know, and manage security risks to defend against unique AI security threats, and embrace MLSecOps for a safer AI-powered world. This includes a broad set of capabilities including AI supply chain security, Auditable Bill of Materials for AI, ML model scanning, signing, attestation and LLM Security.
Join our team to help us solve this critical need of protecting AI!

Role

Protect AI is seeking a Machine Learning Engineer to join our growing team in Berlin. Our engineering team builds and operates ‘Protect AI Platform’ covering everything from AI supply chain security and ML Model Scanning to LLM Security. You will work closely with fellow Software Engineers who bring a variety of skillset to the table including Backend, Frontend, Infra, ML and Security Engineering. You will also get opportunity to work closely with Product Managers, Architects and our CTO.

Responsibilities:

• Build and operate a Machine Learning platform

• Design and develop the platform that will enable innovation, delivery, and operation of ML and AI models throughout Protect AI

• Collaborate across functions (Product Managers, Engineers, and Researchers) to understand requirements and develop solutions

• Act as a technical leader and mentor junior team members

• Continuously improve the software development process, including testing, deployment, and monitoring

Qualifications:

1. Significant experience (4+ years) building and maintaining Machine Learning platforms on Cloud and Kubernetes based infrastructure.

2. Significant experience working with containers and Kubernetes, CI/CD pipelines, and MLOps best practices.

3. Significant expertise in IaC tools (Terraform).

4. Expertise in modern Python (type hints, Pydantic, ..) and at least one strongly typed programming language (Go, Java, C#, C++).

5. Exposure to ML related tools such as PyTorch, TensorFlow, Pandas, Polars, Scikit Learn, and Jupyter.

6. Exposure to ML research and engineering teams developing, deploying, and operating models.

7. Strong communication and problem solving skills.

8. Excited about working in an agile and fast paced environment.
   

What We Offer:

• An exciting, collaborative work environment in a fast-growing startup.

• Competitive salary and benefits package.

• Opportunities for professional growth and development including attending and presenting technical talks at meetups and conferences.

• A culture that values innovation, accountability, and teamwork.

• Opportunities to contribute to our open source projects with thousands of Github stars.

• Work with a team of talented and well-accomplished peers from AWS, Microsoft and Oracle Cloud.

• Work with best in class tools — M2 Macbook Pro, 34” Monitor, modern tech stack and high quality collaboration tools.

• No bureaucracy and legacy systems. You are empowered to innovate and do your best work.
  

Protect AI is an Equal Opportunity Employer. We celebrate diversity and are committed to creating an inclusive environment for all employees.