ML Research Engineer

Triomics; Bengaluru; On-site; INR 800,000 - 1,200,000; Full time

Benefits offered by this job

Unlimited Leave Policy

Lunch Provided at the Office

Flexible Working Hours

Health Insurance

Zomato Meal Benefit

Job summary

A healthcare technology company in Bengaluru seeks a driven ML Research Engineer to develop and prototype innovative NLP solutions. This entry-level position requires 2–3 years of experience in model development, strong Python skills, and familiarity with frameworks like PyTorch and TensorFlow. You will work on cutting-edge LLM architectures, collaborate with teams, and ensure scalable prototypes in a rewarding environment. Enjoy benefits like unlimited leave policy and flexible working hours, alongside great compensation for your contributions.

Qualifications

• 2–3 years of professional R&D experience in NLP or ML research roles.
• Strong understanding of experimental design and statistical significance testing.
• Familiarity with clinical or biomedical NLP is a plus.

Responsibilities

• Rapidly prototype novel LLM architectures and strategies.
• Define evaluation protocols and run ablation studies.
• Build end-to-end POCs demonstrating research applications.

Skills

Strong Python skills

Deep familiarity with PyTorch or TensorFlow

Proficiency in NLP frameworks

Expertise in prompt engineering

Hands-on experience with LLM fine-tuning

Excellent communication skills

Education

Bachelor’s or Master’s degree in Computer Science or related field

Tools

PyTorch

TensorFlow

spaCy

NLTK

HuggingFace Transformers

LangChain

Job description

Position: ML Research Engineer

Type: Full Time

About the Role

We are seeking a driven NLP Research Engineer with 2–3 years of hands‑on experience in model development and prototyping. In this R&D‑focused role, you will explore and implement state‑of‑the‑art LLM architectures, prompt engineering strategies, and LLM training techniques tailored to healthcare and oncology text. Your primary responsibility is to drive proof‑of‑concepts—from rapid iterations to rigorous experimental evaluations on clinical‑text tasks. While our production teams will help with deployment, you’ll collaborate closely to hand off scalable prototypes that demonstrate clear performance gains.

Qualifications

• Bachelor’s or Master’s degree in Computer Science, Computational Linguistics, AI/ML, or a related field
• 2–3 years of professional R&D experience in NLP or ML research roles
• Strong Python skills and deep familiarity with PyTorch or TensorFlow
• Proficiency in working with NLP frameworks (spaCy, NLTK, HuggingFace Transformers, LangChain, vector databases, etc.), deep learning libraries (e.g., PyTorch), and common data preprocessing techniques.
• Expertise in prompt engineering, few‑shot learning, and fine‑tuning for domain‑specific tasks
• Familiarity with building retrieval‑augmented generation (RAG) pipelines.
• Hands‑on experience with LLM fine‑tuning using LoRA and SFT methods
• Solid understanding of experimental design, ablation studies, and statistical significance testing
• Familiarity with clinical or biomedical NLP is a plus
• If you have worked with Direct Preference Optimization (DPO), that’s a plus
• Excellent communication skills for sharing insights in seminars, whitepapers, or publications

Responsibilities

• Model Innovation & Prototyping: Rapidly prototype novel LLM architectures, adapter modules, and prompt‑tuning strategies for tasks such as clinical note summarization, entity extraction, and question answering.
• Experimental Design & Analysis: Define evaluation protocols, run ablation studies, and analyze results to quantify improvements in accuracy, robustness, and domain generalization.
• Proof‑of‑Concept Development: Build end‑to‑end POCs demonstrating how research breakthroughs translate into tangible gains on real healthcare datasets.
• Collaboration & Knowledge Sharing: Work alongside NLP engineers and MLOps teams to hand off production‑ready prototypes; present findings in regular research seminars and contribute to internal or external publications.
• Literature & Trend Monitoring: Continuously survey the latest NLP/ML research, distill key innovations, and propose experiments to test applicability in our clinical environment.

Benefits

• Unlimited Leave Policy – take time off when you need it. We believe in trust, not tracking.
• Lunch Provided at the Office – one less daily decision, one happier employee.
• Flexible Working Hours – we care about output, not clock‑ins.
• Health Insurance – comprehensive coverage for you and your family.
• Zomato Meal Benefit – breakfast and dinner can be ordered when you come in early or leave late, because effort deserves fuel.

Why Join?

• We are revolutionizing a unique industry that has the potential to impact and benefit patients from all over the world – you can create impact at scale.
• We have access to the best computing resources available, including the H100 and A100, among others.
• We have had company‑sponsored workations in Bali, Sri Lanka, and Manali and take pride in our hard‑working yet super fun culture.
• We are working on a few of the most challenging problems in a highly regulated industry, which provides you with an opportunity to solve some of the most interesting things.
• You will get a chance to work with experts from multiple industries, the best in the industry compensation, and to continue building your own (and, of course, new) projects.

Seniority Level

• Entry level

Employment Type

• Full-time

Job Function

• Engineering and Information Technology
• Software Development