Job Description
About the Role:
- Are you passionate about artificial intelligence and its potential to transform industries? Join us at the forefront of legal innovation as we build cutting-edge solutions that are redefining the delivery of legal services across Africa.
- As an AI Engineer, you will be responsible for designing, developing, and deploying AI/ML solutions within our suite of legal tech products. You will work with a multidisciplinary team of software engineers, product managers, and legal professionals to translate legal processes and data into innovative, scalable AI-driven applications.
Key Responsibilities:
- Design, train, and optimize machine learning models for natural language processing (NLP), information extraction, document classification, and legal search.
- Collaborate with product teams to integrate AI solutions into production systems.
- Perform data preprocessing, annotation, and feature engineering on legal datasets.
- Research and implement state-of-the-art AI models aligned with product needs.
- Continuously monitor and improve the performance and scalability of deployed AI models.
- Ensure model interpretability, explainability, and ethical AI compliance.
- Contribute to building data pipelines and tools to support AI development.
Qualifications and Skills:
- Bachelor’s or Master’s degree in Computer Science, Artificial Intelligence, Data Science, or a related technical field.
- 2+ years of hands-on experience in AI/ML engineering, with a strong preference for experience in NLP and transformer-based models.
- Proficiency in Python and machine learning frameworks (e.g., TensorFlow, PyTorch, Scikit-learn).
- Experience with NLP libraries and frameworks such as spaCy, NLTK, Hugging Face Transformers, and OpenAI libraries.
- Demonstrated experience building and deploying Generative AI (GenAI) solutions, including text generation, summarization, and conversational agents.
- Practical experience implementing Retrieval-Augmented Generation (RAG) pipelines by combining LLMs with vector databases to deliver context-aware responses.
- Familiarity with deploying and managing AI models using REST APIs, Docker containers, and cloud-based platforms (AWS, GCP, Azure).
- Solid understanding of data structures, algorithms, and fundamental software engineering principles.
- Experience with vector databases such as Milvus DB, Pinecone, or similar platforms.
- Previous work with large-scale text-based or legal datasets will be considered a significant advantage.
- Excellent analytical, problem-solving, and collaborative teamwork abilities.