Applied AI/ML Lead

Aumni; Mumbai; Hybrid; INR 1,500,000 - 2,500,000; Full time

Job summary

TwinThread is seeking an Applied AI/ML Lead in Mumbai. This role involves designing and managing prompt-based models for NLP in financial services. Candidates should have over 5 years of software engineering experience with strong Python skills, familiarity with cloud platforms like AWS or Azure, and experience in building data pipelines. The position emphasizes collaboration with technical teams and stakeholders to drive innovative solutions in AI/ML. Join us to make a significant impact in the field.

Qualifications

• 5+ years of experience in software engineering.
• Experience with prompt design and implementation.
• Strong programming skills in Python.

Responsibilities

• Design and manage models on LLMs for NLP tasks.
• Conduct research on prompt engineering techniques.
• Build data pipelines and processing workflows.

Skills

Python programming

Prompt design

Data pipeline construction

Cloud platforms (AWS or Azure)

APIs development

MLOps tools

GIT and version control

NLP integration

Education

Formal training or certification in software engineering

Tools

PyTorch

TensorFlow

Java

Spark

Job description

Applied AI/ML Lead

Bengaluru, Karnataka, India

We have an opportunity to impact your career and provide an adventure where you can push the limits of what's possible.

As a Data Scientist Lead at JPMorgan Chase within the Asset and Wealth Management, you are an integral part of an agile team that works to enhance, build, and deliver trusted market-leading technology products in a secure, stable, and scalable way. As a core technical contributor, you are responsible for conducting critical technology solutions across multiple technical areas within various business functions in support of the firm’s business objectives.

Job responsibilities

• Design, deploy and manage prompt-based models on LLMs for various NLP tasks in the financial services domain
• Conduct research on prompt engineering techniques to improve the performance of prompt-based models within the financial services field, exploring and utilizing LLM orchestration and agentic AI libraries.
• Collaborate with cross-functional teams to identify requirements and develop solutions to meet business needs within the organization
• Communicate effectively with both technical and non-technical stakeholders
• Build and maintain data pipelines and data processing workflows for prompt engineering on LLMs utilizing cloud services for scalability and efficiency.
• Develop and maintain tools and framework for prompt-based model training, evaluation and optimization
• Analyze and interpret data to evaluate model performance to identify areas of improvement

Required qualifications, capabilities, and skills

• Formal training or certification on software engineering concepts and 5+ years applied experience
• Experience with prompt design and implementation or chatbot application
• Strong programming skills in Python with experience in PyTorch or TensorFlow
• Experience building data pipelines for both structured and unstructured data processing.
• Experience in developing APIs and integrating NLP or LLM models into software applications
• Hands‑on experience with cloud platforms (AWS or Azure) for AI/ML deployment and data processing.
• Excellent problem-solving and the ability to communicate ideas and results to stakeholders and leadership in a clear and concise manner
• Basic knowledge of deployment processes, including experience with GIT and version control systems
• Familiarity with LLM orchestration and agentic AI libraries
• Hands‑on experience with MLOps tools and practices, ensuring seamless integration of machine learning models into production environment

Preferred qualifications, capabilities, and skills

• Familiarity with model fine‑tuning techniques such as DPO and RLHF.
• Knowledge of Java, Spark
• Knowledge of financial products and services including trading, investment and risk management

Equal Opportunity Statement

We recognize that our people are our strength and the diverse talents they bring to our global workforce are directly linked to our success. We are an equal opportunity employer and place a high value on diversity and inclusion at our company. We do not discriminate on the basis of any protected attribute, including race, religion, color, national origin, gender, sexual orientation, gender identity, gender expression, age, marital or veteran status, pregnancy or disability, or any other basis protected under applicable law. We also make reasonable accommodations for applicants’ and employees’ religious practices and beliefs, as well as mental health or physical disability needs. Visit our FAQs for more information about requesting an accommodation.