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
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 leading sovereign Legal AI. The legally compliant and competent AI helps legal professionals to research legal issues and review, and draft legal documents. The GDPR-compliant Legal AI meets the high professional and data protection requirements for lawyers (§ 203 Penal Code, § 43e Federal Lawyers' Act) and is certified according to ISO 27001, 9001, 27018, 27017, 42001, and BSI C5. The product version Beck-Noxtua is based on the exclusive data of Germany’s leading legal publisher C.H.Beck and Germany’s largest business law firm CMS.

Emerging from a 2017 research project by Dr. Leif-Nissen Lundbæk and Professor Dr. Michael Huth at Oxford University and Imperial College London, the Berlin-based Legal-Tech company, formerly known as Xayn, boasts extensive experience in developing highly efficient GDPR-compliant AI solutions. Strategic partners, including Germany's leading legal publisher C.H.Beck, the High-Performance Computing specialist Northern Data, Germany's largest business law firm and co-initiator of the Legal AI Noxtua CMS, as well as the world's largest law firm Dentons, have invested a total of € 80.7 million in the German startup as part of its Series B funding round.

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.