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

Department Accounts and Finance

JD:  

Accounting Manage and oversee the day-to-day operations of the accounts and finance department.

 Ensure accurate and timely recording of financial transactions.

 Supervise the maintenance of general ledger and subsidiary ledgers.

 Review and approve journal entries and adjustments.

 Perform ledger scrutiny to identity unusual balances and highlight

the same with resolution measures.

Monitor reconciliation of bank accounts, receivables, and payables.

 Managing and maintaining borrowing and liquidity profile of the

Company and assisting CFO in evaluating various fund raising

options.

Preparation of documents relating to fresh borrowings and renewal

of existing credit lines.

Ensuring compliance with Indian GAAP and IND AS as per reporting

requirements of client companies Auditing.

Prepare and present timely and accurate financial statements and

reports to senior management.

 Prepare quarterly Ind AS financial statements as required for Board

&exchange filing.

 Assist in preparation of annual financial statements with disclosures.

 Assist in various audits and data requirements. (Internal, Stat & IFC)

Prepare various information packs as required for management and

board meetings.

Regulatory reporting Prepare and file quarterly RBI XBRL returns.

 Collation of information asked by RBI from time to time and during yearly inspection.

Ensure compliance with TDS, GST & Labour laws.

Skills Strong understanding of accounting principals.

 Knowledge of Indian taxation specially TDS & GST.

 Proficient with MS Office products. (MS Excel, word & outlook)

 Self-motivated and proactive with good communication skills.

 Working knowledge of Oracle fusion will be added advantage.

Strong sales executive profile

Mandatory (Experience 1): Must have 7+ years of hands-on B2B sales experience, preferably selling SaaS products or platforms

Mandatory (Experience 2): Must have experience in doing end to end Hunting / New Business Acquisition, including prospecting, pitching, negotiation, and deal closure

Mandatory (Experience 3): Must have experience of managing the complete sales cycle independently — research, outreach, client meetings, solution pitching, proposal handling, and closing.

Mandatory (Experience 4): Strong communication, presentation, negotiation, and objection-handling skills with experience engaging with mid to senior-level stakeholders

Mandatory (Experience 5): Experience in pipeline building & CRM usage, maintaining accurate forecasts, sales funnel tracking, and performance reporting.

Mandatory (Company): B2B SaaS Product comPreferred Companies

Salesforce, Oracle, Zoho CRM, Keka, Zaggle, Pine Labs (non-prepaid card roles only), Advantage Club, Reward Gatewaypanies

Designation

Business Development Executive

Qualification

MBA

Experience

1 – 3  Years of IT Sales  ( Knowledge of IT Sales & Product )

Location

Pune / Mumbai

Probation period

Up to 3 months

No. of candidate required

1 for Pune / 1 for Mumbai

Responsibility Skills

·        Minimum of 1 to 3 years experience in sales, business development or similar role.

·        Having good knowledge of CRM Software + Sales process.

·        Ability to demonstrate & present CRM and other software to prospect & provide full fledge training to them.

·        Responsible for IT sales(software) for B2B

·        Generate leads through various marketing activities such as cold calling, references and various online sources.

·        Client meetings with Negotiation closure & sales management.

·        The candidate must have good communication skills and convincing skills.

·        Manage all communication with clients over mail & in written.

·        Building Business relationships with current and potential client.

·        Close deal and handle negotiation with the decision makers of prospects.

·        Ability to meet individual & Branch targets.

Mandatory Required Assets

·        Laptop, Bike

Additional Skill

·        Proficiency in CRM software, Market knowledge, Time management and planning skills

·        Good communication skills (verbal and written) and convincing skills

• Source new sales opportunities through inbound lead follow-up and outbound cold calls and emails
• Understand customer needs and requirements
• Proven inside sales experience
• Strong phone presence and experience dialing dozens of calls per day
• Proficient with corporate productivity and web presentation tools
• Excellent verbal and written communications skills
• Strong listening and presentation skills
• Ability to multi-task, prioritize, and manage time effectively

Cogoport Story

Do you prefer to get speeding tickets or parking tickets?

Because at Cogoport we are speeding ahead to do something remarkable for the world. We are trying to solve the Trade Knowledge and Execution Gap, which is currently widening and preventing trade to the tune of $3.4 trillion annually. This Gap has enormous economic as well as human impact and disproportionately hits small and medium businesses globally.

The team at Cogoport is working on developing a new category, the Global Trade Platform, that helps companies discover and connect with appropriate trade partners, optimize shipping and logistics options and costs, and improve cash management and cash flow.

Cogoport is currently in hypergrowth mode. We are proud to have been named an Asia-Pacific High-Growth Company by the Financial Times and an Indian Growth Champion by the Economic Times. We are aiming to reach an annualized revenue of $1 billion (7700 Crores INR)  by this summer and are hiring over 500 additional employees. We are currently hiring in Mumbai, Gurgaon, Chennai and Bangalore.

Cogoport Culture: We have two core values at Cogoport—Intrapreneurship and Customer-centricity. If you share these values and are a hard worker who is willing to take risks (and occasionally get a speeding ticket), you can make a huge impact and propel your career in an endless number of directions with Cogoport.

Cogoport Leadership

https://www.linkedin.com/in/purnendushekhar/">https://www.linkedin.com/in/purnendushekhar/
https://www.linkedin.com/in/amitabhshankar/">https://www.linkedin.com/in/amitabhshankar/

Life at Cogoport: It’s rare to be able to join a company that can give you the resources, support and technology you need to break new ground and see your ideas come to life. You’ll be surrounded by some of the smartest engineers and commercial specialists in India and the Asia Pacific Region.

With huge growth and the right entrepreneurial mindset, comes huge opportunities! So, wherever you join us, you’ll be able to dream, deliver better and brighter solutions, and speed ahead with the possibility to propel your career forward in endless directions as our company continues to grow and expand.

For more insights about the company:  https://cogoport.com/about">https://cogoport.com/about

Why Cogoport?

International Trade can be complicated at times and every day brings new challenges and opportunities to learn.  When we simplify international trade, it empowers and affects every human being on the face of this earth. Seven billion people - one common problem.

As a part of the Talent team at Cogoport, you will get an opportunity to be a part of an industry-wide revolution in the world of shipping and logistics by collaborating with other brilliant minds to resolve real world on-ground challenges. You will have a direct impact on the revenue and profitability growth for the organization.

Areas of Impact for you

• Hands-on management with deep-dive into the details of software design, implementation and debugging.
• Guide your teams in developing roadmaps and systems to drive product growth, then identify, plan, and execute projects to support that growth.
• Manage multiple projects across a wide breadth of technologies, coordinate dependencies, and interactions with the internal teams and external partners.
• Collaborate with stakeholders from across functions to keep the development team in sync with all functions and overall business objectives.
• Develop large multi-tenant applications in Rails.
• Understand Rails best practices and religiously introduce those to the codebase.
• Set up, create and manage strong best practices/architecture to ensure reliable, secure, bug-free, and performant software is released on-time.

Desirable Skills and Experience

• Loves coding.
• 2-4 years of experience building scalable & complex products from scratch.
• Demonstrated ability to build complex scalable technology products.
• Should have prior experience of working with ROR, React, PostgreSQL and cloud infra.
• Understanding scaling strategies for high-traffic Rails applications.
• Understanding O-Auth2 or JWT (Json Web Token) authentication mechanisms.
• Experience in using ActiveRecordSerialize, RSpec and active interaction.
• Knowledge about Asynchronous Networking in Ruby; Refactoring ActiveRecord Models; Background Job processing using Redis and Sidekiq; Writing automated Deployment Scripts using Capistrano, Ansible etc.
• Expertise in Data Science and Machine Learning is a pl
• Expertise in Jenkins, Kubernetes, dockers and cloud technology is a plus.

Cogoport is an equal opportunity employer. We are a welcoming place for everyone, and we do our best to make sure all people feel supported and respected at work.