AI/ML Intern

About the Role

The AI/ML Intern will contribute to the development, testing, and optimization of AI and machine learning models for legal intelligence applications. The role involves working on-site/hybrid at the Hyderabad office.

Key Responsibilities

• Data preprocessing and feature engineering.
• Training and validating AI/ML models.
• Assisting in deploying ML pipelines.
• Analyzing legal data to extract insights.
• Supporting the creation of tools for document classification and search.
• Participating in experiments to improve model performance and reliability.
• Documenting methodologies.
• Preparing reports on findings.
• Collaborating with product and domain experts.

Requirements & Tech Stack

• Proficiency in programming languages commonly used in AI/ML, such as Python.
• Familiarity with libraries like NumPy, pandas, scikit-learn, TensorFlow, or PyTorch.
• Experience with data handling and preprocessing, including working with structured and unstructured text data, cleaning datasets, and performing feature extraction.
• Understanding of natural language processing (NLP) fundamentals, such as tokenization, embeddings, and text classification.
• Solid grasp of transformer fundamental attention, tokenization, embeddings, and how fine-tuning adapts a base model.
• Comfortable in Python, with real GenAI project work experience (e.g., a RAG bot, a parser, an OCR pipeline).
• Ability to read and apply technical research papers.
• Willingness to experiment with different model architectures.
• Ability to communicate findings clearly to technical and non-technical stakeholders.
• Strong analytical thinking and attention to detail.
• Openness to learn about the Indian legal ecosystem and its data challenges.

Qualifications

• Strong foundation in machine learning and data science concepts, including supervised and unsupervised learning, model evaluation, and basic statistics.
• Interest in legal or domain-specific NLP.
• Currently pursuing or recently completed a degree in Computer Science, Data Science, Artificial Intelligence, or a related field.
• Coursework or projects in AI/ML are highly beneficial.
• Comfort working in a collaborative, on-site environment.
• Openness to feedback and mentorship.