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

Responsibilities:

• Assist customers in selecting apparel based on style, fit, and current trends
• Provide personalized styling advice and wardrobe solutions
• Maintain strong product knowledge of fabrics, fits, and collections
• Support customers with size recommendations and outfit coordination
• Ensure visual merchandising standards are followed
• Manage customer queries, trials, and purchase assistance
• Build long-term customer relationships and ensure repeat business
• Track sales, maintain stock awareness, and support store operations

Skills Required:

• Strong sense of style & fashion trends
• Excellent communication & customer service
• Knowledge of fabrics, fits, and apparel categories
• Confident, friendly, and approachable
• Sales & upselling ability
• Good presentation and grooming standards

Core Stack Required: MERN

Location: Remote

Minimum Experience: 3 years

• Knowledge of Node, MongoDB, Express, React.

• Proficient with code versioning tools including Git, Bitbucket.

• Good problem-solving skills.

• Good Knowledge with MySql databases.

• Knowledge and proficiency in HTML, CSS, and JavaScript. (Reactjs is an advantage)

• Experience in payment gateways integration.

• Experience in image manipulation.

• Ability to write and consume SOAP & REST web services.

• Write clean, well-structured code with proper abstraction and reusability.

• Must have good analytical and problem-solving skills.

• Must be a good team player, good learner, and self-starter.

• Ability to work independently and perform under strict deadlines.

• Good to have Experience in creating multilingual websites.

• Experience in API integration i.e. Google, Facebook, Twitter, agora, twilio etc..

Neptune Ubicom Pvt. Ltd. is looking for passionate and talented Java Developers to be a part of our engineering team. We are hiring for Java Developer inspiring to develop into Full-Stack Developer. If you love building high quality, user-friendly web-applications, then you have our attention!

Your day with us!

As a Java Developer, you will design, create and analyse solutions to various project tasks daily. Your day involves brainstorming and knowledge sharing amongst your teammates. You learn and develop your skills by creating unique solutions daily.

Here are a few of your responsibilities?

End to End Software Design which will involve work on Database, working on Back-end and Front-end, documentation, and testing.

• You will be a part of a functional Java developing team
• You will be assigned a project based on your choice and skill set
• You will get the opportunity to work on a complete product development cycle
• You will be involved in the daily scrum and functional discussion meetings with your German counterparts 
• You will be thoroughly guided and supported to develop your technical skillset.

Technologies you get to use:

• Java+Spring (Back-end)
• TypeScript+Angular (Front-end)
• Git, Jira, Confluence, Maven, etc.
• And many more tools that you'll learn on the job!

What makes you a good fit?

• You have a minimum of 2 years of hands-on experience as a Java Developer. 
• You are someone who wants to explore new horizons for their career and develop into a Full Stack role.
• You possess a Bachelor’s Degree in Engineering with specialisation in CS/IT or equivalent
• You have excellent knowledge of Java
• You have great programming skills
• You have demonstrated a passion for coding through various projects, challenges or extra-curricular activities. Having a programmer portfolio is a major plus
• You have the potential to develop a keen understanding of our culture. And to imbibe it, and be the ambassador of our culture.

Why should you join us?

• Embark on an exciting journey with us! 
• Enjoy being part of a great team of engineers, creating innovative new products
• Learn and acquire knowledge of different technologies
• Participate or/and take a lead on various stages of the on-going projects
• Develop your interpersonal skills every day with new challenges
• We do not believe in the crisp cotton shirt is the only productive work attire. Here at NUPL, you dress smart and decent and we will be good to go
• We follow a strict '5 Day' work week policy. Those weekends are for family, friends, and relaxation
• We offer a competitive leave package, a work from home facility, and a flexible work hour policy to help you attain a great work-life balance
• We offer an environment where you can make suggestions and develop your ideas with the team. In a flat hierarchy, you will have your ideas and opinions heard
• We all know how hard it is to focus on work when craving for food, we have an in house open pantry just for this!
• We offer various opportunities to widen your horizon of knowledge and skills. We also offer support for various courses and certifications
• We provide a competitive benefits package.

Interested to know more about us?

We are a multinational IT Company with our offices based in Germany and India. To know more about us and our work culture check out the link below:

https://www.linkedin.com/company/13597515" target="_blank">www.linkedin.com/neptune-ubi/

Joining: ASAP! ; )

Compensation: We offer a competitive package as per the industry standards. The final package is dependent on one’s performance during the recruitment process. We wish all the best for all the aspiring Full Stack Developers applying for this position!

Desired Candidate Profile

• Excellent communication, presentation and organizational skills
• Superlative visual design skills, excellent problem solving skills and the ability to work well in a collaborative team environment
• Understanding of technical constraints, user goals and business requirements that shape a product
• Ability to work confidently with Program Management, Development, Product Planning, Brand, Marketing and Usability
• Proficiency with Figma with an understanding of rapid prototyping platforms such as Invision or Principle
• Experience working within and contributing to a structured design system
• Strong organizational skills and an attention to detail
• User Research experience in a complex space such as enterprise or other business/consumer tools and applications    

Responsibilities:

• Make changes in the existing code base of talent marketplace and implement new features.
• Collaborate with stakeholders that are involved in product design (marketing specialist and product manager).
• Report the progress to other stakeholders.
• Do bug fixes and push the source codes to the live server.
• Be responsive to change requests and feature requests
• Create Unit and Integration tests to ensure the quality of code

Requirements:

• Experience in Node js , MongoDB and Angular.
• Experience in JavaScript development.
• Understanding in MEAN or MERN Stack.
• Experience in the use of project planning tools (JIRA, Version One, GitHub, etc.)
• Excellent written and verbal communication skills (English).
• Understanding of various technologies including applications and databases.
• Able to handle multiple projects and competing deadlines
• Good understanding of overall web design including basic usability, accessibility, industry standards, architecture, and navigation
• Able to handle multiple projects and competing deadlines
• Fluent knowledge of latest HTML/CSS standards and best practices
• Experience building Responsive websites for web, tablet and mobile devices

• We are one of the fastest-growing game development companies with headquarters in
• Bangalore, India. Since our inception in 2017, we have carved a niche in the gaming industry by our highly creative and expert approach to game design, artwork and development. Our two games, ‘Ludo STAR’ and ‘Parchisi STAR’ have already reached 50 million global downloads.
• We have a grand vision of being the “Best Game Developing Company” in the world. We are vastly committed to developing visually appealing and technically advanced games for diverse platforms and game applications that appeal to all ages.
• We offer an autonomous studio consisting of self-motivated teams in a fast decision-making
• environment, allowing you to be truly impactful and develop professionally.
• It’s an exciting time to come aboard, career experience as a lead developer working with leadership teams.