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

About Us:

We are not just an AD agency or a creative agency, we are a Communication Company. Founded in 2014, Moshi Moshi is a young, creative, gutsy, and committed communication company that wants its clients to always Expect the EXTRA from it. Our primary clientele consists of Startups and corporations like Ola, Zoomcar, Mercedes Benz, ITC, Aditya Birla Group, TATA Group, MTV, IHCL, Jaquar, Sobha, Simple Energy, and Godrej amongst others. We have a huge team of creative folks, marketers, learners, developers, and coders who believe Moshi Moshi is an experience rather than a company.

Job Role -Senior Graphic Designer - Branding

Experience Level - 3+ years

Location - Bangalore

Job Description :

This is a full-time on-site role for a Senior Graphic Designer in Bengaluru. The Senior Graphic Designer will be responsible for creating graphics, designing logos, developing branding, and typography. They will be expected to interpret design briefs, collaborate with internal and external stakeholders, and deliver high-quality designs.

Qualifications :

● Expertise in Graphics, Graphic Design, Logo Design, and Branding

● Advanced skills in Typography

● Hands-on experience with design software like Adobe Creative Suite or Sketch

● Strong understanding of visual communication principles

● Ability to interpret client needs and design briefs into actionable design solutions

● Experience in managing multiple projects and meeting deadlines

● Excellent interpersonal and communication skills to collaborate with other team members and stakeholders

● Experience in advertising or creative agencies is a plus

● 3 + creative design experience

Please attach your portfolio/Behance link to the resume while applying

Role & responsibilities

• Develop and maintain server-side applications using Go Lang.
• Design and implement scalable, secure, and maintainable RESTful APIs and microservices.
• Collaborate with front-end developers to integrate user-facing elements with server-side logic
• Optimize applications for performance, reliability, and scalability.
• Write clean, efficient, and reusable code that adheres to best practices.

Preferred candidate profile

• Minimum 5 years of working experience in Go Lang development.
• Proven experience in developing RESTful APIs and microservices.
• Familiarity of cloud platforms like AWS, GCP, or Azure.
• Familiarity with CI/CD pipelines and DevOps practices

Job Title - Mern Stack Trainer

Skills Required - Java, Python, React, Nodejs, MongoDB

Experience - 3-4 Years

Package- 6-10LPA

Location - Coimbatore.

Job Responsibility:

• Manage digital advertising campaign activities: Setup, traffic, manage and optimize online ad campaigns, campaign modification as per request, campaign scheduling, targeting and monitoring its performance.
• Day to day campaign optimization by focusing on conversion and other performance metrics.
• Reporting of the campaign performance metrics on timely basis to various stakeholders.
• Sustaining present clients for further business opportunities.
• Shall take lead in achieving monthly delivery target on existing publishers
• Strategizing execution of different campaigns across various verticals.
• Interaction with agencies/ad networks to plan and run campaigns.

Job Requirements:

• Prior experience of managing CPL/CPI/CPR/CPT/CPS Campaign Delivery & Optimization in US/RU/CA/UK/UAE/SA/JP/KR preferred.
• Working knowledge of tracking platform (hasoffers, affise etc)
• Excellent English Communications skills – written & verbal.

We are looking for an experienced Lead Java Developer to join our team and take a leadership role in driving the design, development, and delivery of highquality software solutions. A Senior Java Lead Developer will be responsible for leading a team of developers, providing technical guidance, and ensuring the successful execution of projects. Should have a strong background in Java development, excellent leadership skills, and a track record of delivering complex software projects on time. System Architect: Should have Architect the system before for a large scale solution Understanding of various java design pattern and use the pattern which is more appropriate Understanding of various micro-service design pattern and use the pattern which is more appropriate Has worked on DDD with CQRS Has designed solution for multi tenant application. Technical skills: Used latest version of java and good understanding of it Skilled in spring boot, spring MVC Basic Skill: 10+ years of experience in development. Good communication skills in English. Testing: Skilled in using Junit Security: Good understanding of application security Understanding of DDOS and best practice for that Good understanding of rate limiting API Good understanding about spring security or any thing better Database: Great understanding of MySQL Experience in extensively using hibernate Debugging: Understanding about Logging Skilled in debugging production issue

 About Freespace* (a.k.a. Workplace Fabric)  

Workplace Fabric develops solutions that makes workplaces intelligent. Our solutions help increase productivity of office workers while helping to lower the cost of real estate. Our Flagship 

product Freespace is installed in over 40 cities around the world and serves over 30,000 people every day. Starting with the first deployment in March 2016 we have rapidly grown with our solution being rolled out globally while continuing to stay cash positive. With successes at several blue-chip clients Workplace Fabric has carved a unique leadership position in workplace management and employee engagement SaaS solutions. 

*To find out more, follow us on social media – LinkedIn & Facebook @ workplace-fabric 

What is on our offer?  

• An absolutely action packed, fun, and demanding workplace lets our employees to thrive in a team-based environment with passion, fun, pride and teamwork as our core values. 
• We can offer you career opportunities to challenge, develop and exceed your aspirations with training and development throughout your career. 
• Life is never dull at Freespace as there is always something to learn, constantly pushing you to develop your skills and take on the next exciting challenge. 

 Profile Brief:  

You will be a creative Frontend Engineer with specialist React and visualization skills in our fast growing Analytics Team, creating dynamic dashboards to drive our client’s business insights.  

Working closely as an active team member, having ownership & accountability to develop the analytics stories, in delivering timely projects through Scrum. Working with product owners and team leads to clarifying requirements, challenge assumptions in order to ensure successful executions and push the client analytics to its best.  

Key Skills and Experience  

Required Skills: JavaScript (ES6+), React, amCharts, D3.js (or industry equivalent) 

Good to have: SQL, TypeScript, AWS, Scrum, IoT 

Experience: 4-6 years  

• A front-end and data visualization expert with 4+ years of experience  

• Ability to design widget or component based, scalable analytics dashboards 
• Expert in creating excellent user experience through quick loading dashboards and pixel-perfect design 
• Implement dashboard solutions with interactive and dynamic visualization to tell the story. 
• Hands-on and well-versed with at least one of React (preferred), Angular or Vue  
• Data processing and manipulation expert 

• Master in JavaScript with excellent understanding of asynchronous patterns  
• Comfortable with of ES6+, TypeScript & emerging web technologies  
• Able to easily comprehend and write SQL queries  
• Hands-on programmer  

Page Break 

Key Behaviours  

• Have an open mind to turn business requirements into attractive yet practical visualisations. 

• Be passionate about understanding the business problem and using data to tell the stories in delivering solutions which drive client decisions. 
• Contribute to the success of the team through active participation, team work and owning project deliverables. 
• Contribute to high standards through personal delivery of data quality checks, code reviews and documentation in work and team practices. 
• Clear thinker with a systematic problem-solving approach, seeing a project through from conception to finished product. 
• Unafraid to own mistakes and be willing to challenge and feedback to enhance the team processes and solutions.