Senior AI Engineer (m/f/d) | Legal AI Tech Start-up | Full Remote

Permanent employee, Full-time · Berlin, Mobile/Hybrid

Our offer for you

You will join our AI Team led by Felix (Head of AI), working closely with a group of approximately 5 AI experts. This highly collaborative team focuses on pushing the boundaries of generative AI, natural language processing, and privacy-preserving machine learning legal solutions.

HIRING MANAGER

Felix, our Head of AI, will guide you through your journey at Xayn. With deep expertise in AI systems, Felix leads with a passion for innovation and a collaborative approach, ensuring every team member thrives.

OUR BENEFITS

Working hours: Flexible working hours: Full-time (32- 40h/week).
Vacation: 26 days + December 24th & 31st off
Remote: 100% remote work possible (given a German residence), other countries upon request
Discounts: e.g. Urban Sports Club Membership
Equipment: Laptop (Lenovo or Mac), second screen, keyboard etc.

Your responsibilities

Develop and deploy scalable machine learning systems, focusing on privacy-preserving AI
Build and optimize NLP applications using transformer models like BERT, GPT, and RAG
Create workflows using agentic frameworks for automated context-aware decision making
Optimize models for resource efficiency using techniques like quantization, pruning and distillation.
Using open-source ML frameworks (e.g., TensorFlow, PyTorch)
Develop AI pipelines, including retrieval-augmented generation systems
Conduct benchmarking and evaluation of large language models (LLMs)
Work with vector databases and manage data processing pipelines

Our Tech Stack

Programming Languages: Python
Frameworks: LangChain, LangGraph, LlamaIndex
Libraries: HuggingFace, Transformers, NumPy, Pandas, Pydantic, FastAPI, OpenAI & PyTorch
Deployment Tools: Docker
Cloud Infrastructure: AWS, GCP, Azure
MLOps: CI/CD pipelines for AI models, including monitoring and version control
Vector Databases: ElasticSearch, Qdrant, Pinecone
Ticket System: Atlassian JIRA
Repository: Github
CI/CD System: GitHub Actions
Documentation: Confluence
Communication: Slack
Office Application: MS365

About you

Residence & Work Permit: Eligible to work in Germany or within the EU.
Language: English proficiency at C2 level.
Experience: in AI development with at least 3 successfully deployed projects
NLP & Generative AI: Expertise in developing and deploying NLP and generative AI models
Open-Source Models: Solid experience with open-source models (e.g., LLAMA, Mistral, Cohere)
RAG: Knowledge in building RAG systems
Programming: Strong Python skills and experience with AI pipelines
Data: Familiarity with data processing, filtering, and augmentation

Optional:

Fine-tuning of pre-trained LLMs using custom datasets for domain adaptation
Proficiency in vector database management
Experience in deploying custom AI solutions with privacy considerations

Apply for this job

About us

The AI startup Xayn develops Noxtua, Europe’s first sovereign Legal AI. Noxtua helps lawyers analyze and draft legal documents and research legal questions while being legally competent and compliant. Hosted in the EU, Noxtua meets high standards of confidentiality in handling client data, professional secrecy (e.g. Section 203 German Criminal Code, Section 43e German Federal Code for Lawyers), and European data protection. Noxtua is powered by specialized proprietary AI models. The models are trained with high-quality legal data provided by the Legal AI Alliance which the tech startup Xayn initiated with the international business law firm CMS. This makes Noxtua the secure, independent, and specialized European Legal AI.
The Berlin-based AI company Xayn was born out of research at Oxford University and Imperial College London by Dr Leif-Nissen Lundbæk and Professor Michael Huth. Founded in 2017, Xayn’s academic vision remains, with a workforce comprised of approx. 30% PhDs. The startup has received investment funding of 19.5 million EURO from Global Brain Corporation, KDDI Open Innovation Fund, Earlybird, and Dominik Schiener.

Apply for this job

Your application!

We appreciate your interest in Xayn AG. Please fill in the following short form. Should you have any difficulties in uploading your files, please contact us by mail at jobs@xayn.com.