Senior ML Engineer
Uttar Pradesh, Lucknow, India
EBC Technologies
- Design and build AI-powered search models that improve retrieval and ranking of legal documents.
- Implement retrieval-augmented generation (RAG) workflows using pre-trained LLMs (e.g., OpenAI GPT-4).
- Fine-tune LLMs for legal use cases where necessary (experience with custom LLM training is a strong plus).
- Improve search quality through relevance testing, feedback loops, and query understanding.
- Research and implement any new techniques for improving search result relevancy.
- Build pipelines to ingest, chunk, and vectorize legal texts (case law, statutes, etc.).
- Create and maintain indexes in Vector Databases, supporting fast and relevant results.
- Maintain an evolving legal search index by ingesting new documents on a weekly basis.
- Deploy ML models into production using Azure cloud infrastructure.
- Develop REST APIs (with FastAPI or Flask) to expose model functionality to the application layer.
- Monitor and optimize latency, scalability, and reliability of deployed solutions.
- Work closely with product managers and full-stack engineers to ship ML-backed features.
- Participate in design reviews and own technical decisions around AI architecture.
- Track and improve system performance using user feedback, telemetry, and experimentation.
- ML/NLP: Python, PyTorch/TensorFlow, Hugging Face, Azure OpenAI APIs
- Vector Search: Azure AI Search (primary), experience with FAISS, Pinecone or Elasticsearch a plus
- Deployment: Azure (App Services, Azure Functions, Blob Storage, Key Vault)
- Data Processing: Pandas, NumPy, spaCy, NLTK
- APIs: REST APIs built with FastAPI or Flask
- 5+ years of experience in machine learning, NLP, or AI-based search systems.
- Strong knowledge of vector search, document embeddings, and retrieval techniques.
- Experience building and scaling RAG pipelines with LLMs.
- Proficiency with Azure AI Search for document indexing and search optimization.
- Demonstrated ability to deploy models to production and build robust APIs.
- Familiarity with search ranking algorithms (BM25, hybrid search, learning-to-rank).
- Experience working with document-heavy datasets in legal, academic, or enterprise domains.
- Experience with fine tuning models and creation of datasets used in fine tuning.
- On-site position for Lucknow, India.
- Background in legal tech, contract analysis, or legal document retrieval.
- Exposure to open-source search frameworks like Elasticsearch or OpenSearch.
- Knowledge of observability, logging, and system performance profiling.
* Salary range is an estimate based on our AI, ML, Data Science Salary Index 💰
Tags: API Development APIs Architecture Azure Elasticsearch Engineering FAISS FastAPI Flask GPT GPT-4 LLMs Machine Learning ML models Model deployment NLP NLTK NumPy OpenAI OpenSearch Open Source Pandas Pinecone Pipelines Python PyTorch RAG Research spaCy TensorFlow Testing
More jobs like this
Explore more career opportunities
Find even more open roles below ordered by popularity of job title or skills/products/technologies used.