AI/ML Engineer (LLM Fine tuning)

Role Overview: We are seeking a highly skilled Senior AI/ML Engineer with extensive experience in document classification, natural language processing (NLP), and Large Language Models (LLMs). The ideal candidate will lead the development of our Document AI framework, specifically focusing on interpreting complex property and financial documents. You will be responsible for the end-to-end lifecycle of machine learning models, from fine-tuning LLMs for specialized extraction tasks to deploying scalable APIs in a production environment.

Key Responsibilities

• Model Development & Fine-tuning: Lead the selection, architecture, and fine-tuning of LLMs for high-accuracy document classification and data extraction from unstructured sources.
• End-to-End Extraction Pipelines: Design and optimize workflows to extract structured data from diverse documents such as GST certificates, Udyam registrations, FSSAI licenses, and property-related legal documents.
• Performance Evaluation: Conduct rigorous end-to-end analysis and result metric evaluation (e.g., Precision, Recall, F1-score) to ensure model reliability across production batches.
• OCR Integration: Collaborate with engineering teams to integrate central OCR services and refine extraction keys for specific business release cycles.
• Production Deployment: Build and maintain high-performance REST APIs and microservices using Python (FastAPI/Flask) within AWS environments.
• Advanced AI Tasks: Implement techniques for resolving name variations, identifying corporate hierarchies, and normalizing unstructured text from SMS and bureau records.

Required Skills & Qualifications

• Experience: 2-5 years of hands-on experience in AI/ML engineering, with a proven track record in document intelligence or specialized NLP.
• LLM Expertise: Demonstrated experience fine-tuning LLMs (e.g., Llama, Mistral, BERT-based architectures) for classification and information extraction tasks.
• Technical Stack: Strong proficiency in Python, including data handling libraries like Pandas and NumPy, and ML frameworks such as PyTorch or TensorFlow.
• Production Knowledge: Familiarity with CI/CD pipelines, MLOps, Docker, Kubernetes, and cloud platforms like AWS (specifically Lambda or EC2-based API deployments).
• Database Proficiency: Solid understanding of SQL (MySQL) and NoSQL databases for handling large-scale document metadata.
• Education: Bachelors or Masters degree in Computer Science, AI, Mathematics, or a related field

KreditBee is committed to maintaining a transparent, fair, and secure recruitment process. We have become aware of fraudulent individuals and entities falsely claiming to represent KreditBee and offering fake employment opportunities.

Please note that KreditBee never asks candidates to pay any money for registration, training, certifications, security deposits, or any other purpose at any stage of the recruitment process. All official recruitment communications are sent through verified company email domains, primarily @kreditbee.in, @krazybee.com, and @kbnbfc.in. Candidates are advised to verify the sender's email address before responding to any recruitment-related communication. KreditBee does not conduct recruitment through unofficial email addresses or request payments.

If you receive a suspicious job offer, interview invitation, or recruitment-related communication claiming to be from KreditBee, do not make any payment or share sensitive personal or financial information. We encourage you to verify the authenticity of the communication through our official careers page or by contacting us at hr@kreditbee.in

Disclaimer: KreditBee shall not be responsible for any loss, damage, or claims arising from fraudulent job offers or recruitment communications made by unauthorized individuals or entities falsely representing the company. Candidates should exercise due diligence and verify the authenticity of all recruitment-related communications before taking any action.This job description is intended to outline the general nature and key responsibilities of the position. It is not intended to be an exhaustive list of all duties, responsibilities, or qualifications associated with the role. The responsibilities and qualifications described may be subject to change, and other duties may be assigned as needed. Employment is at-will, meaning the employee or the employer may terminate the employment relationship at any time, with or without cause, and with or without notice.

Data Utilization Disclaimer:
By applying for this position, you acknowledge and agree that any personal data you provide may be used for recruitment and employment purposes. The data collected will be stored and processed in accordance with our privacy policy and applicable data protection laws. Your information will only be shared with relevant internal stakeholders and will not be disclosed to third parties without your consent, unless required by law.