11+ JAX Jobs in Mumbai | JAX Job openings in Mumbai

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

Design, implement, and improve the analytics platform

Implement and simplify self-service data query and analysis capabilities of the BI platform

Develop and improve the current BI architecture, emphasizing data security, data quality

and timeliness, scalability, and extensibility

Deploy and use various big data technologies and run pilots to design low latency

data architectures at scale

Collaborate with business analysts, data scientists, product managers, software development engineers,

and other BI teams to develop, implement, and validate KPIs, statistical analyses, data profiling, prediction,

forecasting, clustering, and machine learning algorithms

Educational

At Ganit we are building an elite team, ergo we are seeking candidates who possess the

following backgrounds:

7+ years relevant experience

Expert level skills writing and optimizing complex SQL

Knowledge of data warehousing concepts

Experience in data mining, profiling, and analysis

Experience with complex data modelling, ETL design, and using large databases

in a business environment

Proficiency with Linux command line and systems administration

Experience with languages like Python/Java/Scala

Experience with Big Data technologies such as Hive/Spark

Proven ability to develop unconventional solutions, sees opportunities to

innovate and leads the way

Good experience of working in cloud platforms like AWS, GCP & Azure. Having worked on

projects involving creation of data lake or data warehouse

Excellent verbal and written communication.

Proven interpersonal skills and ability to convey key insights from complex analyses in

summarized business terms. Ability to effectively communicate with multiple teams

Good to have

AWS/GCP/Azure Data Engineer Certification

Responsibilities:

• Design multi-tier Elasticsearch architecture (hot/warm/cold/frozen) with ILM.
• Architect Active-Active/Passive HA and DR across production and disaster recovery data centers.
• Define and implement observability standards, data schemas, naming conventions, and ILM policies per domain.
• Develop and manage ingestion pipelines using Logstash, Beats, and Elastic Agent
• Parse and enrich structured/unstructured logs; implement scalable and fault-tolerant pipelines.
• Deploy and configure Elastic APM agents for Java/.NET/Python services
• Ensure end-to-end distributed tracing, span enrichment, and service dependency mapping
• Optimize trace sampling, error logging, and business transaction visibility
• Design end-to-end observability solution using Elastic stack: log analytics, metrics, traces, APM, and ML.
• Integrate machine learning for anomaly detection.
• Oversee implementation of distributed tracing and service mapping. Ensure that every critical banking transaction can be traced end-to-end.
• Drive adoption of observability standards (SLOs/SLIs, tagging, dashboards).

Rejolut Technology Solutions Pvt Limited

Accounts & Finance

Experience : 3 Years TO 7 Years

Location : Navi Mumbai (Juinagar)

Maintain general ledger and ensure accurate posting of transactions

Prepare and process invoices, payments, and payroll

Manage accounts receivable and accounts payable

Reconcile bank statements and company accounts regularly

Assist in preparing monthly, quarterly, and annual financial statements

Handle GST, TDS, and other statutory filings (as applicable)

Coordinate with auditors for annual audits

Monitor budgets and assist in financial planning and analysis

Support project costing and IT-specific accounting (e.g., SaaS revenue recognition, R&D cost tracking)

Ensure compliance with accounting principles and internal controls

1. Design cloud infrastructure that is secure, scalable, and highly available on AWS, Azure and GCP
2. Work collaboratively with software engineering to define infrastructure and deployment requirements
3. Provision, configure and maintain AWS, Azure, GCP cloud infrastructure defined as code
4. Ensure configuration and compliance with configuration management tools
5. Administer and troubleshoot Linux based systems
6. Troubleshoot problems across a wide array of services and functional areas
7. Build and maintain operational tools for deployment, monitoring, and analysis of AWS, Azure Infrastructure and systems
8. Perform infrastructure cost analysis and optimization

Skill Requirements:

• Proven software development experience in PHP.
• Good knowledge of relational databases, version control tools and of developing web services.
• Experience in common third-party APIs (Google, Facebook, Ebay etc).
• Passion for best design and coding practices and a desire to develop new bold ideas

(Leading Health & Nutrition based Startup in Mumbai)

• Creating Campaigns in Google ads, Facebook marketing & other online marketing Platforms 
• Keyword Research & Monitoring
• Writing Ad copy/description
• Basic knowledge of Google Sheets
• Must be proficient in the English Language
• Monitors the latest trends in social media, including advertising formats, channels and technologies in order to improve campaign performance and provide recommendations 
• Compile data across several platforms and create weekly/monthly reports, including analysis for insights, optimizations and future strategy development
• Displays organizational capabilities to track progress, execution and consistency of social advertising campaigns
• Running Email/SMS Campaigns
• Demonstrates understanding of and ability to facilitate and manage forecasting, budgeting and pacing, campaign creation and optimization
• Coordinate with Ad Agencies & take regular updates
• Coordinate with content team to get the required creatives
• Coordinate with production team to ensure inventory is in place before running campaigns

Job ID: RP1050

Job Location: Mumbai or Gurugram

Expected Joining: within two weeks

Job Requirement:

• 4 to 7 years of experience in the lending domain.
• Proficiency with selenium using Java or Python
• Proficiency in Linux, network database, distributed system.
• Develop test plans, identify quality risks and propose reasonable solutions
• Organize and execute test tasks

About Us:

Mobile programming LLC is a US-based digital transformation company. We help enterprises transform ideas into innovative and intelligent solutions, governing the Internet of Things, Digital Commerce, Business Intelligence Analytics, and Cloud Programming. Mobile programming is SAP Preferred Vendor, Apple Adjunct Partner, Google Empaneled Mobile Vendor, and Microsoft Gold Certified Partner.

• Primavera planner & Scheduler
• Exp - 5 to 8 years
• Exp in EPC (Engineering procurement and construction) projects
• 1 year contract
• Location – Mumbai
• Language: English, Hindi
• Good coordinating and communicating skill