AI Engineer - LLM Agents in TypeScript

🌟 About You

Are you fascinated by LLM agents and its potential to transform industries? Are you eager to expand your hands-on experience in designing, implementing, and improving real-world AI applications? Do you thrive in a fast-paced environment where innovation and problem-solving are the norm? If so, we’d love to hear from you!

🚀 About Omnilex

Omnilex is a young dynamic AI legal tech startup with its roots at ETH Zurich. Our passionate interdisciplinary team of 10+ people is dedicated to empowering legal professionals in law firms and insurance companies by leveraging the power of AI for legal research and answering complex legal questions. You can check out our demo at https://ailegis.pro/demo.

🛠️ Your Responsibilities

As an AI Engineer, your primary focus will be on improving our LLM Agent and Retrieval-Augmented Generation (RAG) system, a cornerstone of our legal research platform. Your tasks will include:

• Designing and optimizing LLM Agents to perform various tasks such as summarization, classification, text generation, reasoning, and data extraction with reliability
• Enhancing the data retrieval for better performance, accuracy, and scalability with methods such as query processing, hybrid search (keywords + semantic embeddings), semantic reranking, relevance scoring and knowledge graphs
• Designing and implementing additional tests and metrics on legal datasets for separate parts of the LLM-Agent and RAG pipelines
• Integrating new data sources
• Collaborating with our interdisciplinary team to align technical implementations with legal and user requirements.
• Communicating your work and findings to the team for continuous feedback and improvement.

📌 Qualifications

✅ Minimum qualifications

• Degree in Computer Science, Data Science, AI, or a related field.
• Knowledge and strong intuition in machine learning fundamentals (especially NLP), statistics, probability, and metrics
• Experience with LLM agents flows and RAG
• Ability to learn quickly and adapt to the challenges of a dynamic startup environment.
• TypeScript experience or strong interest to learn
• Excellent problem-solving and communication skills.
• Proficiency in English
• Genuine interest in learning about the Swiss and other legal systems 🧑‍⚖️
• Availability full-time, 80% and above may also be considered for exceptional candidates. On-site in Zurich at least one day per week.

🎯 Preferred qualifications

• You have a Swiss work permit or EU/EFTA Citizenship.
• German-speaking (most of the Swiss legal data is in German and we are expanding to Germany)
• Familiar with our stack: Azure/NestJS/NextJS
• Familiar with relevant TypeScript libraries such as for document parsing (e.g. PDF.js, pdf-parse), LLM Agents (e.g. LangChain, LangGraph), testing (e.g. Jest, LangSmith) and edge computing (e.g. Transformers.js)
• Extracurricular coding activities such as competitive programming, hackathons, Kaggle competitions, or research labs

🤝 Benefits

• Direct involvement in cutting-edge AI projects with immediate real-world applications.
• A chance to further your valuable expertise in LLM Agent and RAG systems in a legal tech context.
• Work with a young, energetic, and diverse team of engineers, designers, business people and legal professionals.
• Opportunity to contribute to a product that is revolutionizing legal research.
• Compensation between CHF 7’000 and CHF 10’000 per month + ESOP (employee stock options), depending on experience and skills.

We’re excited to hear from candidates who are passionate about AI and eager to make an impact in the legal tech space. Apply today by pressing on the Apply button.