Jupitice.com - AI Engineer - LLM Models

We are seeking an experienced AI Engineer to join our team and spearhead the development of cutting-edge AI solutions with a focus on Generative AI and applications in the legal domain. The ideal candidate will have a strong background in building and training AI / ML models from scratch, deploying them into production, and working with advanced NLP, LLMs, and vector databases. If you have previously created AI models, including custom Large Language Models (LLMs) or domain-specific models, you will be an excellent fit for this role.

Key Responsibilities

• Design, develop, and train custom AI models, including Generative AI and NLPbased models tailored for the legal domain.
• Implement techniques for fine-tuning and optimizing large language models for domain-specific use cases.
• Build models from scratch where required, ensuring scalability, accuracy, and efficiency.
• Design and implement AI-driven systems using Python and frameworks such as FastAPI and Django for robust API development.
• Develop and integrate vector databases for semantic search, embeddings, and retrieval-augmented generation (RAG).
• Manage data pipelines, including preprocessing and cleansing of unstructured text (legal documents, contracts, case law, etc.).
• Work with SQL and PostgreSQL databases for structured data storage and retrieval.
• Develop generative solutions for document summarization, legal drafting, knowledge extraction, and automated reasoning.
• Leverage modern transformer architectures (e.g., GPT, BERT, LLaMA, Falcon, etc.) to build advanced capabilities.
• Collaborate with product, legal, and engineering teams to understand requirements and deliver solutions in an Agile environment using tools like Jira and GitHub.
• Mentor junior engineers and contribute to knowledge sharing within the team.
• Stay updated with the latest advancements in AI, LLMs, RAG, and Generative AI technologies and apply them to real-world use cases.
• Propose innovative solutions to improve efficiency, accuracy, and scalability of existing systems.

Preferred Skills & Qualifications

Good To Have

(ref : hirist.tech)

Skills Required

Github, Postgresql, Jira, Sql, Tensorflow, Django, Nlp, Pytorch, Docker, Mysql, FastAPI, Python