11+ JAX Jobs in India

Review Criteria

• Strong Senior Data Scientist (AI/ML/GenAI) Profile
• 5+ years of experience in designing, developing, and deploying Machine Learning / Deep Learning (ML/DL) systems in production
• Must have strong hands-on experience in Python and deep learning frameworks such as PyTorch, TensorFlow, or JAX.
• 1+ years of experience in fine-tuning Large Language Models (LLMs) using techniques like LoRA/QLoRA, and building RAG (Retrieval-Augmented Generation) pipelines.
• Must have experience with MLOps and production-grade systems including Docker, Kubernetes, Spark, model registries, and CI/CD workflows

Preferred

• Prior experience in open-source GenAI contributions, applied LLM/GenAI research, or large-scale production AI systems
• Preferred (Education) – B.S./M.S./Ph.D. in Computer Science, Data Science, Machine Learning, or a related field.

Job Specific Criteria

• CV Attachment is mandatory
• Which is your preferred job location (Mumbai / Bengaluru / Hyderabad / Gurgaon)?
• Are you okay with 3 Days WFO?
• Virtual Interview requires video to be on, are you okay with it?

Role & Responsibilities

Company is hiring a Senior Data Scientist with strong expertise in AI, machine learning engineering (MLE), and generative AI. You will play a leading role in designing, deploying, and scaling production-grade ML systems — including large language model (LLM)-based pipelines, AI copilots, and agentic workflows. This role is ideal for someone who thrives on balancing cutting-edge research with production rigor and loves mentoring while building impact-first AI applications.

Responsibilities:

• Own the full ML lifecycle: model design, training, evaluation, deployment
• Design production-ready ML pipelines with CI/CD, testing, monitoring, and drift detection
• Fine-tune LLMs and implement retrieval-augmented generation (RAG) pipelines
• Build agentic workflows for reasoning, planning, and decision-making
• Develop both real-time and batch inference systems using Docker, Kubernetes, and Spark
• Leverage state-of-the-art architectures: transformers, diffusion models, RLHF, and multimodal pipelines
• Collaborate with product and engineering teams to integrate AI models into business applications
• Mentor junior team members and promote MLOps, scalable architecture, and responsible AI best practices

Ideal Candidate

• 5+ years of experience in designing, deploying, and scaling ML/DL systems in production
• Proficient in Python and deep learning frameworks such as PyTorch, TensorFlow, or JAX
• Experience with LLM fine-tuning, LoRA/QLoRA, vector search (Weaviate/PGVector), and RAG pipelines
• Familiarity with agent-based development (e.g., ReAct agents, function-calling, orchestration)
• Solid understanding of MLOps: Docker, Kubernetes, Spark, model registries, and deployment workflows
• Strong software engineering background with experience in testing, version control, and APIs
• Proven ability to balance innovation with scalable deployment
• B.S./M.S./Ph.D. in Computer Science, Data Science, or a related field
• Bonus: Open-source contributions, GenAI research, or applied systems at scale

Role overview

- Hands-on Experience: 3-5 years of experience in automation testing,

specifically for API and UI applications.

- Strong Coding Skills: Proficiency in Java or JavaScript (for

Playwright) with a strong understanding of object-oriented programming

concepts.

- Automation Frameworks: Experience in creating and maintaining automation

frameworks using Selenium and Playwright (preferred), with Rest Assured for

API testing.

- API Testing: Experience in designing and executing automated tests for

RESTful APIs using tools like Postman, Rest Assured, or similar.

- Version Control: Strong experience with version control systems like Git.

- CI/CD Integration: Experience with integrating automated tests into CI/CD

pipelines (Jenkins, GitLab CI, etc.).

- Agile Methodology: Comfort working in an Agile environment with fast

iterations and continuous feedback.

- Problem-Solving: Strong debugging, problem-solving, and analytical skills to

troubleshoot issues effectively.

- Communication Skills: Excellent verbal and written communication skills to

clearly report test results and collaborate with cross-functional teams.

- Collaboration & Mentoring: Ability to collaborate with various stakeholders and

mentor junior team members, fostering a culture of learning and growth within

the team.

- Passion for Test Automation: A self-motivated individual with a passion for

quality and test automation, always looking for innovative solutions and

improvements.

- Right Attitude: A proactive attitude towards taking on challenging work and

responsibilities, with a positive approach to tackling complex tasks.

What would you do here

- Test Automation: Write, execute, and maintain automated tests using

Selenium, Playwright, and Rest Assured.

- Mentorship: Guide and mentor junior engineers in the team, helping them

improve their automation skills and providing technical support.

- Collaboration: Work closely with cross-functional teams including developers,

product managers, and other QA engineers to understand requirements and

deliver automated tests accordingly.

-- Continuous Improvement: Drive improvements in test automation processes

and practices, ensuring scalability, reusability, and high performance of test

suites. Responsibility & Ownership: Take full responsibility for the automation tasks assigned, from design to execution and reporting.

- Adaptability: Thrive in a fast-paced, agile environment and adapt quickly to

changing requirements.

- Defect Reporting & Resolution: Collaborate with the development team to

troubleshoot and resolve issues detected during test automation.

- Test Coverage & Reporting: Ensure comprehensive test coverage and provide

clear, actionable test results and metrics

Review Criteria

• Strong Product Manager Profiles
• 7+ years of product management experience, of which 2+ years in healthcare, pharmaceutical, life sciences, or AdTech domains
• Must have built or scaled products involving Data Science, Machine Learning, or AI
• Must have experience working end-to-end on product lifecycle — strategy, roadmap, development execution, stakeholder alignment, user research, product optimization, and adoption (0 to 1 product experience is preferred)
• Hands-on experience collaborating with engineering, data science, design, supply teams, and demand-side teams on parallel product initiatives
• Must have a strong understanding of demand-side and supply-side mechanisms, programmatic advertising, data intelligence products, or marketplace platforms
• Experience in companies serving Healthcare Professionals (HCPs), Pharma, Life Sciences, or HealthTech advertising is a must
• Top/Well known Product companies (preferably in HealthTech)
• CTC includes 20% variable
• HealthTech exposure is a must (current or past experience)
• Its an IC role

Preferred

• Experience working on AI-driven features such as predictive models, segmentation, personalization, or automated optimization

Job Specific Criteria

• CV Attachment is mandatory
• What is your preferred location — Noida or Mumbai?
• If you’re based in Mumbai, are you comfortable traveling to the Noida office for one week each month?
• Are you available for an in-person interview for one of the rounds?
• Which HealthTech company(ies) you have worked for?

Role & Responsibilities

We are seeking a visionary and technically savvy Principal Product Manager to lead the strategy, development, and growth of our Identity Resolution products, including the HCP Graph, probabilistic and deterministic identity solutions, and data partner integrations. This role sits at the intersection of data, product, and healthcare advertising, giving you the opportunity to shape how company identifies, unifies, and activates HCP audiences across the ecosystem. You will define the future of identity in healthcare marketing and play a pivotal role in building products that maximize both privacy compliance and advertiser ROI.

Key Responsibilities-

Product Strategy & Vision:

• Define and evangelize the vision for company's identity resolution products and HCP data graph.
• Build a product roadmap that balances technical innovation, market differentiation, and business impact.
• Conduct competitive research and explore emerging identity technologies to ensure company remains at the cutting edge of HCP data solutions.

Product Development & Execution:

• Lead end-to-end product lifecycle: ideation, requirement definition, development, launch, and iteration.
• Collaborate with engineering and data science teams to design scalable identity solutions leveraging first-party, third-party, and probabilistic signals.
• Author high-quality PRDs, user stories, and acceptance criteria, ensuring clarity for cross-functional teams.
• Drive integration strategy with data partners, DSPs, and other activation channels to expand company's reach and addressability.

Data & Privacy Expertise:

• Develop solutions that unify HCP identities across platforms while maintaining strict compliance with privacy regulations like HIPAA, GDPR, CCPA, and industry-specific rules.
• Ensure high identity match rates, data quality, and robust governance frameworks.
• Explore advanced techniques such as probabilistic linking, email-based resolution, and graph modeling to continuously improve audience unification.

User Experience & Enablement:

• Collaborate with UX designers to simplify complex workflows for internal teams and clients managing HCP audiences.
• Provide clear product documentation, training, and enablement materials to maximize adoption.

Performance & Optimization:

• Define metrics for success (match rate, segment addressability, HCP coverage, platform adoption) and continuously optimize solutions.
• Use data-driven insights to iterate on products, identify gaps, and enhance identity resolution accuracy and reliability.

Ideal Candidate

• Bachelor’s or Master’s in Computer Science, Engineering, Data Science, or related field.

Experience:

• 7+ years in product management, with significant experience owning data-centric products, preferably identity resolution, CDPs, DMPs, or large-scale data systems in AdTech/MarTech.
• Proven track record in data integration, audience unification, or identity resolution products.

Skills & Expertise:

• Strong knowledge of identity systems: deterministic and probabilistic matching, identity graphs, email-based resolution, hashed identifiers, device IDs, and consent management.
• Deep understanding of healthcare and HCP data ecosystems is a plus.
• Ability to translate complex technical solutions into clear product strategy and business impact.
• Exceptional stakeholder management, communication, and leadership skills.
• Strong analytical mindset with experience in SQL, data modeling, or analytics.
• Comfort operating in a fast-paced, high-growth environment and leading cross-functional initiatives.

Nice-to-Haves:

• Hands-on experience with machine learning for audience modeling or lookalike targeting.
• Familiarity with privacy-enhancing technologies, clean rooms, and data encryption.
• Experience building and scaling HCP identity graphs or multi-source data unification platforms.

POSITION: QA Engineer

JOB OVERVIEW:

We are looking for a QA Engineer to assess software quality through Manual and Automated Testing. You will be responsible for finding and reporting software bugs and glitches. You should have a keen eye for details and must be competent in executing test cases. You should also possess excellent communication and problem-solving skills and should be passionate about quality ensuring that our products, applications, and systems work correctly.

WHAT YOU WILL DO:

● Testing life cycle SDLC STLC.

● Analysis of the requirement Design & Implement functional test-cases.

● Reviewing software requirements and preparing test scenarios.

● Executing tests on software usability.

● Analyzing test results on database impacts, errors or bugs, and usability.

● Preparing reports on all aspects related to the software testing carried out and reporting to the design team.

● Interacting with clients to understand product requirements.

● Participating in design reviews and providing input on requirements, product design, and potential problems.

● Ensure complete test -coverage, interact with Product management development team & different stakeholders.

● Involve in Test-plan, Test scenario Test strategy review. Participate in the defect triage call, use the existing Tool & built the new Test script

● Hands-on API testing with REST Assured, Postman, Swagger.

● Should be having a good Analytical Skill.

● Should be expert in SQL severRedshift & web service testing.

● Expert in Agile & waterfall methodology & feature file writing.

● Assist in setting up & built the new Test- environments.

● Helping in new processes & procedures.

● Candidate must have good understanding of Automation Framework

● Test Execution (Expert) - Deep dive into large volumes of data comparison validation

● Test Framework (Expert) - Solid understanding of Test Phases, Type of testing to be undertaken & be able to perform them. Risk gate assessments for entry & exit criteria's. Test outputs - plan, cases, results, sign off reports.

REQUIRED SKILLS

● Must have strong experience working in the Software Testing field throughout the career across all phases of software development life cycle.

● SAAS application testing experience

● Must have exposure as well as hands on experience of manual as well as performance testing for various web based, mobile based and desktop-based Software Products and Application projects.

● Must have strong hands-on experience of various types of testing including end to end functional testing,UIUX, Performance, Security, Architecture, Websites and Performance testing etc.

● Must have led various software testing projects end to end including Test Plans & Test Cases

preparation, Effort Estimation, Test Reporting, Project Coordination activities for various projects.

● Knowledge of Database testing over MySQL or other relational non-relational databases is a plus.

● Strong experience with Automation tools including Selenium and others for testing Web basedRESTAPIsWeb ServicesSchedulersBatches & Event driven applications.

● Knowledge of at least one Programming language & coding on an Automation framework is plus.

● Must have familiarity to leading tools for requirement management, test case management, bug

management and version control.

● Should have strong documentation and verbal communication skills and excellent aptitude.

● Good to have: JIRA, Lean FT, Jenkins

Role Overview: As an Business Development Executive, you'll convert leads, qualify prospects, and engage customers to boost sales. Your efforts will directly impact our mission to empower professionals worldwide.

Joining: Immediate

Interview Mode: Face-to-Face

Key Responsibilities:

• Convert leads from various marketing channels
• Create and execute short- and long-term sales plans
• Identify cross-selling and upselling opportunities
• Leverage customer referrals to expand the pipeline
• Build strong customer relationships

Sales & Financial Counsellor – IVF (Delhi/Gurgaon/Noida/Ghaziabad)

Key Responsibilities:

Financial Counselling:

Provide face-to-face financial counselling to patients, helping them understand the costs involved in IVF treatment and offering customized payment plans.

Conversion: Effectively communicate the benefits of our IVF packages to potential patients, with the goal of converting inquiries into confirmed treatment plans.

Patient Grievance Handling: Address and resolve patient grievances promptly, ensuring patient satisfaction throughout their journey with us.

Patient Relationship Management: Build and maintain strong relationships with patients, ensuring they feel supported and informed throughout their treatment process.

Collaboration: Work closely with the medical team to ensure a seamless patient experience and address any financial or administrative concerns.

Key Qualifications:

Experience: 2-5 years of experience in financial counselling, preferably in a healthcare or IVF setting.

Communication Skills: Excellent verbal and written communication skills, with the ability to explain complex financial concepts in a simple and empathetic manner.

Sales Skills: Proven track record in sales or counselling roles, with the ability to convert inquiries into confirmed sales.

Problem-Solving Skills: Strong ability to handle patient grievances and ensure high levels of patient satisfaction.

Empathy: A caring and empathetic approach towards patients, understanding their emotional and financial concerns.

Keywords:

Financial Counsellor

IVF Financial Advisor

Patient Relationship Manager

Healthcare Sales

IVF Counsellor

Patient Grievance Handling

Healthcare Financial Consultant

IVF Package Sales

Patient Satisfaction

Fertility Treatment Advisor

Why Join Us?

Impactful Role: Play a key role in helping couples achieve their dreams of parenthood by guiding them through the financial aspects of IVF treatment.

Supportive Environment: Work in a caring and supportive environment where patient satisfaction is the top priority.

Career Growth: Opportunities for professional growth and development within the organization.

Location: Ahmedabad (Temporary WFH)

Designation: QA Engineer

Experience: Fresher to 3 Years

Salary: As per market standards

Responsibilities

1-Should have played a vital role in Setting up Automation Frameworks, Processes, and Standards.

2-Expert proficiency in automation scripting using Java or other scripting languages.

3-Experience in using automation tools like Selenium/Webdriver, Sikuli, Junit/JMeter, etc.

4-Good expertise in System, Regression, Smoke, Sanity, Ad-Hoc, Browser Compatibility, and Interface testing.

5-Prepare test case for Functional, Non-Functional, Integration and Usability, UI.

6-Experienced with a defect and bug tracking tools like Bugzilla, JIRA, etc.

7-Hands on experience in Test Planning, Test designing, and Test Execution, Defect Tracking & Defect Reporting.

8-Experience working in an agile development environment and scrum methodologies.

9- Involvement in setting up Test Environments, Requirements Management, Requirement Analysis.

10-Good understanding of database.

11-Should have experience in API testing.

12- Should have knowledge of payment gateway testing.

13- Good Knowledge of Performance Testing, Regression Testing, DataBase testing, Android Testing.

Qualifications

Education: B.E/B.Tech BCA/MCA

Designation- Customer Service Representative

Job Location- Kaikondrahalli, Sarjapur Road, Bangalore

Educational & Skill-Set Qualifiers:

Education - Graduation/HSC qualified,

Excellent verbal and written communication skills.

Good Interpersonal skills, and numerical and analytical ability.

Decision-making skills.

Languages -

English & Hindi mandatory, Kannada/Tamil/preferred

6 days a week and the week-off will be on weekdays.

Specific Responsibilities:-

• Handle Inbound and Outbound calling for customer service.
• Identify and assess customers’ needs to achieve satisfaction.
• Build sustainable relationships and trust with customer accounts through open and interactive
•  
• Provide accurate, valid, and complete information by using the right methods/tools.
• Handle customer complaints, provide appropriate solutions and alternatives within the time limits
• Conduct follow-ups to ensure complete resolution of issues.
• Keep records of customer interactions and process customer accounts.
• Follow communication procedures, guidelines, and policies.
• Take the extra mile to engage customers.

Remunerations- Fixed CTC (20k to 27k for 0-4 yrs of exp. respectively ) + 30% incentive

Lead/Principal Engineer – Android

We are looking for talented engineers who are passionate about using software engineering to make kids love to learn and redefine learning@home at BYJU’S. You will have an opportunity to work in a highly motivated fast-paced environment in a fastest-growing Ed-Tech company. You will be building platforms and services to be used across various products. You will have the opportunity to create a significant impact on our systems, our business and most importantly on the life of millions of our students as we take on challenges that will reshape the education industry. You will operate at the intersection of diverse disciplines - Backend, Frontend, Big Data, ML, Computer vision and Multiple programming languages. You will build solutions for Byju’s core learning Apps, co-branded platforms, new launches, product development & innovations and personalized learning experiences.

We are looking for smart people with excellent technical skills and a can-do attitude. Candidates should be eager to learn, flexible to pivot when needed, savvy enough to thrive in a dynamic environment and should possess a growth mindset.

• Leads more than one projects end-to-end and collaborates across functions. Drives planning, estimation and execution.
• Manages stakeholder expectations and offers scalable, reliable, performant and easy to maintain solutions.
• Consistently takes well thought technical/design decisions.
• Develops expertise in more than one area and shares knowledge with others. Able to mentor/train people in areas which are new to them.
• Drives people to solve engineering challenges.
• Enjoys high respect of Tech and other cross functional teams.
• Demonstrates effective communication with project team, management and internal/external clients as necessary.
• Surfaces both technical and non-technical team challenges and helps resolve them.
• Champion for SDLC best practices and high-quality standards.
• Significantly contributes in hiring high performance candidates.

• Expert in JAVA, Kotlin, Reactive programming, Asynchronous programming and Dependency Injection. Experience with unit-tests and automation frameworks.
• Familiarity with any architectures like MV*(MVP/MVVM/MVI).
• Strong focus on high-quality mobile UI experiences.
• Operates independently with almost no oversight.
• Is able to apply domain expertise to think critically and make wise decisions for the team, taking into account tradeoffs and constraints.
• Communicates tech decisions through design docs and tech talks.
• Has delivered multiple projects with end-to-end engineering ownership.
• Keeps track of new technology/tools and embraces them as necessary
• 7+ years of experience in a product-driven organization.

Technologies we're working with

• MVP, JAVA and Kotlin, Realm, Firebase Real-time Database, RxJava,Kotlin Coroutines, Dagger and Retrofit.

Note- WFH currently

About the Role:

We are looking for passionate and highly motivated Technologist who has delivered rock solid software solutions for industry leading enterprises with lean and agile teams. We are looking for an engineer with a strong technical background, thorough understanding of testing methodologies, a passion for quality, and a desire to make a positive change in our products. 

Roles & Responsibilities

• Work on FieldEZ product development
• Perform requirements analysis, design, coding, unit testing, reviewing and implementation
• Work closely with the manager to prioritize the requirements based on technical dependencies for product development
• Provide mentoring and coaching to junior team members
• Team management and customer management

Skills & Experience

• 5 to 7 years of experience in Information Technology
• Deep understanding and hands on experience in Core Java, J2EE, JavaScript, HTML, Spring, Spring Security, Spring Boot, Microservices, Hibernate, JPA, CSS3, Angular 2+, Jquery, SQL, MySQL, PostgreSQL, Scripting, Stored Procedures, Jenkins, Git
• Experience with REST APIs
• Work on Design patterns, UML Design (Any UML tool is fine), Application and DB clustering.
• Very good understanding of database (MySQL or Oracle), knowledge in query performance optimization
• Knowledge of development lifecycle, standards, approaches, utilities, and environment
• Exposure to responsive / progressive web development would be an added advantage.

Educational Qualification

B.Tech/ B.E or M.Tech/M.E (Computer Science/Information Technology/Electronics & Communication/ Information Science) , MCA from a reputed School and University.