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

WeAssemble is looking for an Email Marketing Executive to support and build marketing communications. We are looking for an intermediate-level marketing professional with a track record of delivering predictable and sustainable results in a fast-paced environment.

This position collaborates closely with the Digital team, Marketing Data Operations, IT for marketing automation, and stakeholders from multiple business units.

Responsibilities

●       Control and direct the email marketing channel, as well as the related analysis

●       Create engagement strategies based on an understanding of customer needs and intent.

●       Support the execution of automated lead generation programs

●       Develop and manage digital templates (emails, landing pages, programs) within the marketing automation platform.

●       Marketing Automation Platform expert

●       Meet or exceed internal SLAs and seamlessly coordinate across time-zones as needed

●       Work in partnership with Marketing Automation Segment Liaison to optimize campaign requests to meet business goals.

●       Take ownership of marketing technology platforms

●       Analyse and report on the performance of channels and campaigns.

●      Make it easier to report on marketing impact.

●       Work closely with internal and cross-functional stakeholders (Design, Brand, Legal, Business Units, etc.) to design and approve email campaigns.

●       Ensure that email permissions are followed.

●       Implement best practices for email marketing.

●       Aid in the evaluation of new innovative opportunities to increase engagement and/or results.

●       Keep up to date with changes/developments in the IT sector through self-learning, training, and research

Desired Candidate Profile

●       3+ years outreach experience with working on a marketing automation platform such Woodpecker, Apollo.io or we connect

●       Must be able to demonstrate a high level of marketing automation

●       Excellent communication skills written, verbal, presentation

●       Proven ability to work in a collaborative environment as well as being a motivated, self-starter

●       Keep up-to-date on email and marketing automation best practices with regard to marketing automation capabilities, testing and projected results.

●       Report on critical email and marketing automation campaign metrics

●       Understand lead scoring and nurturing programs as it pertains to execution of email and marketing automation campaigns

●       Support the execution of automated lead generation programs

●       Build and execute test strategies, including A/B multi-variant testing, to drive qualified traffic and revenue-building activities

●       Monitor the campaigns and make necessary changes to generate the lead. 

Role: Lead AI Engineer

Exp: 3-6 Years

CTC: 35.00-40.00 LPA

Work Mode :WFH

Mandatory Criteria (Can't be neglected during screening) :

• Need Excellent Communication skills as the company is dealing with US Clients also

• 3+ years in AI development, with experience in multi-agent systems, logistics, or related fields.

• Proven experience in conducting A/B testing and beta testing for AI systems.

• Hands-on experience with CrewAI and LangChain tools.

• Should have hands-on experience working with end-to-end chatbot development, specifically with Agentic and RAG-based chatbots. It is essential that the candidate has been involved in the entire lifecycle of chatbot creation, from design to deployment.

• Should have practical experience with LLM application deployment.

• Proficiency in Python and machine learning frameworks (e.g., TensorFlow, PyTorch).

• Experience in setting up monitoring dashboards with tools like Grafana, Tableau, or similar.

• Proficiency with cloud platforms (AWS, Azure, GCP)

Job Title: Software Dev Engineer II

Location: Bangalore

Duration: 8 Months    

Job Type: Contract      

Work Type: Onsite 

Job Description: 

The top 3 responsibilities:

·        Requirement gathering & Coding

·         

Mandatory Requirements:

·        2+ years in Core Java, DSA.

·        Strong coding experience

·        Candidate based in Bangalore will be preferred

·        Quick joiners will be preferred

·        Candidate should have good communication skills

·         

Education or Certification:

·        Bachelors in CS, IT or BSC in CS

 Notice: Immediate Joiners only

Shift timings: 9 AM- 6 PM IST some flexibility will be required after 6 PM IST

Location: Bangalore

Relevant Exp: 2+ years

Mode: All 5 days WFO

3-6 years of experience in Functional testing with good foundation in technical expertise.

Experience in Capital Markets domain is MUST.

Exposure to API testing tools like SoapUI and Postman.

Well versed with SQL

Hands on experience in debugging issues using Unix commands.

Basic understanding of XML and JSON structures

Knowledge of Finesse is good to have

Skills Required:

- Good experience with programming language Python
- Strong experience in Docker.
- Good knowledge with any of the Cloud Platform like Azure.
- Must be comfortable working in a Linux environment.
- Must have exposure into IOT domain and its protocols ((Zigbee & BLE ,LoRa,Modbus)
- Must be a good team player.
- Strong Communication Skills

About the role:

We are seeking a highly skilled Azure DevOps Engineer with a strong background in backend development to join our rapidly growing team. The ideal candidate will have a minimum of 4 years of experience and has have extensive experience in building and maintaining CI/CD pipelines, automating deployment processes, and optimizing infrastructure on Azure. Additionally, expertise in backend technologies and development frameworks is required to collaborate effectively with the development team in delivering scalable and efficient solutions.

Responsibilities

• Collaborate with development and operations teams to implement continuous integration and deployment processes.
• Automate infrastructure provisioning, configuration management, and application deployment using tools such as Ansible, and Jenkins.
• Design, implement, and maintain Azure DevOps pipelines for continuous integration and continuous delivery (CI/CD)
• Develop and maintain build and deployment pipelines, ensuring that they are scalable, secure, and reliable.
• Monitor and maintain the health of the production infrastructure, including load balancers, databases, and application servers.
• Automate the software development and delivery lifecycle, including code building, testing, deployment, and release.
• Familiarity with Azure CLI, Azure REST APIs, Azure Resource Manager template, Azure billing/cost management and the Azure Management Console
• Must have experience of any one of the programming language (Java, .Net, Python )
• Ensure high availability of the production environment by implementing disaster recovery and business continuity plans.
• Build and maintain monitoring, alerting, and trending operational tools (CloudWatch, New Relic, Splunk, ELK, Grafana, Nagios).
• Stay up to date with new technologies and trends in DevOps and make recommendations for improvements to existing processes and infrastructure.
• ontribute to backend development projects, ensuring robust and scalable solutions.
• Work closely with the development team to understand application requirements and provide technical expertise in backend architecture.
• Design and implement database schemas.
• Identify and implement opportunities for performance optimization and scalability of backend systems.
• Participate in code reviews, architectural discussions, and sprint planning sessions.
• Stay updated with the latest Azure technologies, tools, and best practices to continuously improve our development and deployment processes.
•  
• Mentor junior team members and provide guidance and training on best practices in DevOps.

Required Qualifications

• BS/MS in Computer Science, Engineering, or a related field
• 4+ years of experience as an Azure DevOps Engineer (or similar role) with experience in backed development.
• Strong understanding of CI/CD principles and practices.
• Expertise in Azure DevOps services, including Azure Pipelines, Azure Repos, and Azure Boards.
• Experience with infrastructure automation tools like Terraform or Ansible.
• Proficient in scripting languages like PowerShell or Python.
• Experience with Linux and Windows server administration.
• Strong understanding of backend development principles and technologies.
• Excellent communication and collaboration skills.
• Ability to work independently and as part of a team.
• Problem-solving and analytical skills.
• Experience with industry frameworks and methodologies: ITIL/Agile/Scrum/DevOps
• Excellent problem-solving, critical thinking, and communication skills.
• Have worked in a product based company.

What we offer:

• Competitive salary and benefits package
• Opportunity for growth and advancement within the company
• Collaborative, dynamic, and fun work environment
• Possibility to work with cutting-edge technologies and innovative projects

4-5 years of overall experience in software development.
- Experience on Hadoop (Apache/Cloudera/Hortonworks) and/or other Map Reduce Platforms
- Experience on Hive, Pig, Sqoop, Flume and/or Mahout
- Experience on NO-SQL – HBase, Cassandra, MongoDB
- Hands on experience with Spark development,  Knowledge of Storm, Kafka, Scala
- Good knowledge of Java
- Good background of Configuration Management/Ticketing systems like Maven/Ant/JIRA etc.
- Knowledge around any Data Integration and/or EDW tools is plus
- Good to have knowledge of  using Python/Perl/Shell

Please note - Hbase hive and spark are must.

• Advise and discuss with the team on the architecture and requirements.
• Bring your experience for mobile User Experience.
• Set up actions to ensure high level of quality and satisfaction
• Build and maintain efficient and reliable code.
• Support continuous improvement by investigating alternatives and technologies.

• Bachelor’s Degree in Informatics Engineering or similar
• Minimum of 4+ years work experience in similar roles
• Strong knowledge in Front-end frameworks (React Native)
• Strong knowledge in the mobile language (swift, objective-c, java, kotlin)
• Experience in unit testing and TDD.
• Advance knowledge in using testing frameworks ( Karma, Jest, Appium, Cypress ).
• Experience with Agile Methodologies (Scrum, Kanban)
• Experience in CI/CD
• Having a coaching profile, good communication skills is crucial to ensure good exchange and sharing of information within and outside of the train
• Being able to provide based on the audience synthetic or detailed information is mandatory
• Being curious to continuously learn and being a leader in your domain of expertise
• Immediate or very low availability
• Fluent in English - written and spoken (mandatory)