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

Position: Team Lead – Data Engineering

Client: NucleusTeq

We are looking for an experienced Team Lead / Architect – Data Engineer to design, architect, and lead scalable cloud-based data platforms.

Key Responsibilities:

·     Design and architect scalable, high-performance data solutions.

·     Lead and mentor a team of data engineers.

·     Build and optimize large-scale data pipelines.

·     Drive architectural decisions and technical strategy.

·     Ensure data quality, governance, and security best practices.

·     Collaborate with cross-functional teams for end-to-end delivery.

Required Skills:

·     Strong hands-on experience in Java & Python.

·     Strong expertise in SQL and Big Data technologies.

·     Hands-on experience with Google Cloud Platform (GCP).

·     Experience with Apache Airflow for workflow orchestration.

·     Strong understanding of distributed systems and data architecture.

·     Experience in CI/CD pipelines (mandatory).

·     Experience in data warehousing and ETL processes.

·     Excellent leadership and stakeholder management skills.

Good to Have:

·     Experience with Dataflow (Apache Beam).

·     Experience designing enterprise-level data platforms.

·     Exposure to DevOps best practices.

Qualifications & Experience: -

• Must have a bachelor’s degree in computer science or a related field (or equivalent degree and experience).
• 2+ years providing customer-facing technical support (ITOM applications).
• Good Communication Skills.

To be successful in this role, we need someone who has:

• Ability to comprehend, modify server-side object-oriented JavaScript, Unix Shell scripting and Windows scripting and PowerShell.
• Ability in server administration across OS platforms (Windows/Unix). Including understanding of user permission levels, domain configuration, group policy objects and execution of SUDO Policies.
• Understanding of protocols such as HTTP, TCP, FTP/SFTP, SOAP and REST, Experience with troubleshooting tools such as Wireshark, Traceroute.
• Strong troubleshooting/root cause isolation skills
• Demonstrated creative problem-solving approach and strong analytical skills
• Must be proficient with analyzing log files and standard debugging concepts.
• Familiarity with tools & practices of the trade such as incident management, knowledge base, defect & escalation management.
• Excellent communication skills (verbal and written).

Preferred: - 

• Solid understanding of object-oriented programming skills (Java strongly preferred)
• A fundamental understanding of ITOM, CMDB and ITIL business process. 
• Experience providing web development support is highly desirable.
• Previous experience working with the ServiceNow platform (Asset Management, Orchestration, Discovery, Mid Server, BSM Map & Service Mapping).
• Understanding of JavaScript or any other scripting language with logical approach for given problem statement.
• Previous experience in software development(or) software consulting.
• Experience supporting and troubleshooting infrastructure level issues.

We are looking for an enthusiastic and skilled SDE 2 React Web Developer to join its dynamic engineering team. As an SDE 1, the developer will be responsible for building and maintaining user-facing features for the web platform using React. The developer will work closely with cross-functional teams to deliver high-quality, scalable, and efficient applications that offer an exceptional experience to Revv's users.

Responsibilities

• Develop responsive, user-friendly, and high-performance web applications using React.js and related technologies.
• Collaborate with product managers, designers, and other engineers to deliver new features and improve existing ones.
• Write clean, maintainable, and testable code, ensuring a high level of code quality.
• Optimize applications for maximum speed and scalability.
• Troubleshoot and debug application issues, ensuring a smooth user experience.
• Stay up-to-date with the latest industry trends and best practices in web development.
• Participate in code reviews and contribute to continuous improvement within the team.

Requirements

• Bachelor’s degree in Computer Science, Engineering, or a related field.
• Proficiency in React.js, JavaScript (ES6+), HTML5, and CSS3.
• Experience with state management libraries like Redux or Context API.
• Familiarity with version control systems, primarily Git.
• Knowledge of RESTful APIs and integrating them into front-end applications.
• Strong understanding of web performance optimization techniques.
• Basic experience with testing frameworks like Jest, Mocha, or Enzyme.
• Strong problem-solving skills and ability to work independently and as part of a team.

We are looking for a Node.js Developer responsible for managing the interchange of data between the server and the users. Your primary focus will be the development of all server-side logic, definition, and maintenance of the central database, and ensuring high performance and responsiveness to requests from the front-end. You will also be responsible for integrating the front-end elements built by your co-workers into the application. Therefore, a basic understanding of front-end technologies is necessary as well.

Responsibilities:

• Integration of user-facing elements developed by front-end developers with server-side logic
• Writing reusable, testable, and efficient code
• Design and implementation of low-latency, high-availability, and performant applications
• Implementation of security and data protection
• Integration of data storage solutions
• Performance tuning, improvement, and optimization of the application
• Ensuring high performance of applications and providing support

Skills and Qualifications:

• Strong proficiency with JavaScript and Node.js
• Knowledge of Node.js and frameworks available for it such as Express
• Understanding the nature of asynchronous programming and its quirks and workarounds
• Good understanding of server-side templating languages such as Jade, EJS
• Basic understanding of front-end technologies, such as HTML5, and CSS3
• Understanding accessibility and security compliance
• User authentication and authorization between multiple systems, servers, and environments
• Integration of multiple data sources and databases into one system
• Understanding fundamental design principles behind a scalable application
• Understanding differences between multiple delivery platforms, such as mobile vs. desktop, and optimizing output to match the specific platform
• Creating database schemas that represent and support business processes
• Implementing automated testing platforms and unit tests
• Proficient understanding of code versioning tools, such as Git
• Familiarity with continuous integration
• Familiarity with AWS
• Experience with Mongo DB/Dynamo DB

Education and Experience:

• Bachelor’s degree in Computer Science, Engineering, or a related field
• 2+ years of experience as a Node.js Developer or similar role

Job Location: Pune 

Experience: 4- 5 years

Functional Area - IT Software - Application Programming , Maintenance

Role Category : Programming & Design

Requirement / Job Description:

 Core Skills:

 Strong experience of Core Java (1.7 or higher), OOPS concepts and Spring framework (Core,     AOP, Batch, JMS)

 Demonstrated design using Web Services (SOAP and REST)

 Demonstrated Microservices APIs design experience using Spring, Springboot

 Demonstrable experience in Databases like MySQL, PostgreSQL, Oracle PL/SQL development etc

Strong coding skills, good analytical and problem-solving skills

Excellent understanding of Authentication, Identity Management, REST APIs, security and best practices

 Good understanding of web servers like Tomcat Apache, nginx or Vertex/ Grizzly, JBoss etc

 Experience in OAuth principles

 Strong understanding of various Design patterns

Other Skills:

  Familiarity with Java Cryptography Architecture (JCA)

 Understanding of API Gateways like Zuul, Eureka Server etc..

 Familiarity with Apache Kafka, MQTT etc.

Responsibilities:

 Design, develop, test and debug software modules for an enterprise security product

 Find areas of optimization and produce high quality code

 Collaborate with product managers and other members of the project team in requirements specification and detailed engineering analysis.

 Collaborate with various stake holders and help bring proactive closure on the issues

 Evaluate various technology trends and bring in the best practices

 Innovate and come out of the box solutions

Adapt, thrive and deliver in a highly evolving and demanding product development team

Come up with ways to provide an improved customer experience

Job Description

We are looking for a Verbal Ability trainer to take classes for students preparing for various competitive exams like RRB, BANKING, SSC, etc.

ROLES & RESPONSIBILITIES:

• Taking classes and doubt sessions on verbal ability for Banking, SSC and RRB Exams.
• Solving doubts of students individually as well as in a group session.
• preparing questions and test series for Bank exams, SSC and RRB.
• Conduct Doubt Clearing sessions for area-specific problems.
• Mentoring students to handle exam-specific challenges.
• Create lesson plans and strategies that are custom designed for students in group.
• Prescribe follow-up work to be undertaken by students to revise and reinforce concepts and skills dealt with in classroom discussions.
• Be prepared and equipped for both offline and online sessions.
• Be part of promotional and awareness activities and campaigns.
• Stay updated on exam and test related patterns, schedules, subjects, eligibility, difficulty level, etc.

SKILLS REQUIRED

• Exposure/Updated knowledge of the Banking, RRB and SSC Exams is essential.
• Must be passionate about teaching.
• Strong planning and organizational skills.
• Excellent analytical and problem-solving abilities.
• Patience and resilience.
• A high degree of diplomacy.
• Great conflict resolution skills.
• Fluent in English or Hindi
• Those candidates who have already been successful in Banking exams/SSC/RRB will get preference.
• Must be bilingual. Knowledge of Hindi is a must. Additional knowledge in Tamil, Telugu, or Malayalam is an added advantage.

• Designing and overseeing all aspects of our digital marketing department including our marketing database, email, and display advertising campaigns.
• Developing and monitoring campaign budgets.
• Planning and managing our social media platforms.
• Preparing accurate reports on our marketing campaign’s overall performance.
• Coordinating with advertising and media experts to improve marketing results.
• Identifying the latest trends and technologies affecting our industry.
• Evaluating important metrics that affect our website traffic, service quotas, and target audience.
• Working with your team to brainstorm new and innovative growth strategies.
• Overseeing and managing all contests, giveaways, and other digital projects.

• Analyze organic search landscapes to discover opportunities that may drive valuable traffic
• Help keep our pages SEO optimized for mobile users and search performance
• Perform extensive keyword research and product listing audits to put forth optimized content regularly
• Grow into an SEO leader, taking additional responsibilities as and when needed
• Conduct competitive site analysis
• Monitor SEO impact with key metrics
• Develop and implement link building strategies
• Work with the editorial team to drive SEO traffic in content creation through other digital marketing channels such as the blog
• Recommend changes to website architecture, content, linking, and other factors impacting SEO positions for target keywords

• Has a minimum of 2 years of SEO experience Offpage, Onpage, Technical SEO
• Is passionate and flexible to get innovative to boost organic search; should not accept theories at face value and should prefer testing, exploring to prove the effectiveness of tactics
• Is adept with SEO suits and tools such as Google Webmaster Tools, Analytics, SEMRush, Ahrefs, among others
• Has extensive knowledge in link building, keyword research, content creation, and on-page optimization (is a plus)
• Has prior experience of managing a team or is willing to take up the responsibility of managing a team and is a team player