11+ JAX Jobs in Bangalore (Bengaluru) | JAX Job openings in Bangalore (Bengaluru)

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

Company name: JPMorgan (JPMC)

Job Category:
 Predictive Science

Location: Parcel 9, Embassy Tech Village, Outer Ring Road, Deverabeesanhalli Village, Varthur Hobli, Bengaluru

Job Schedule: 
Full time

JOB DESCRIPTION

JPMC is hiring the best talents to join the growing Asset and Wealth Management AI team. We are executing like a startup and building next-generation technology that combines JPMC unique data and full-service advantage to develop high impact AI applications and platforms in the financial services industry. We are looking for hands-on ML Engineering leader and expert who is excited about the opportunity. 

As a senior ML and GenAI engineer, you will play a lead role as a senior member of our global team. Your responsibilities will entail hands on development of high-impact business solutions through data analysis, developing cutting edge ML and LLM models, and deploying these models to production environments on AWS or Azure. 

You'll combine your years of proven development expertise with a never-ending quest to create innovative technology through solid engineering practices. Your passion and experience in one or more technology domains will help solve complex business problems to serve our Private Bank clients. As a constant learner and early adopter, you’re already embracing leading-edge technologies and methodologies; your example encourages others to follow suit. 

 Job responsibilities 

• Hands-on architecture and implementation of lighthouse ML and LLM-powered solutions 

• Close partnership with peers in a geographically dispersed team and colleagues across organizational lines 

• Collaborate across JPMorgan AWM’s lines of business and functions to accelerate adoption of common AI capabilities 

• Design and implement highly scalable and reliable data processing pipelines and deploy model inference services. 

• Deploy solutions into public cloud infrastructure 

• Experiment, develop and productionize high quality machine learning models, services, and platforms to make a huge technology and business impact 

 Required qualifications, capabilities, and skills

• Formal training or certification on software engineering concepts and 5+ years applied experience

• MS in Computer Science, Statistics, Mathematics or Machine Learning. 

• Development experience, along with hands-on Machine Learning Engineering 

• Proven leadership capacity, including new AI/ML idea generation and GenAI-based solutions 

• Solid Python programming skills required; with other high-performance language such as Go a big plus 

• Expert knowledge of one of the cloud computing platforms preferred: Amazon Web Services (AWS), Azure, Kubernetes. 

• Experience in using LLMs (OpenAI, Claude or other models) to solve business problems, including full workflow toolset, such as tracing, evaluations and guardrails. Understanding of LLM fine-tuning and inference a plus 

• Knowledge of data pipelines, both batch and real-time data processing on both SQL (such as Postgres) and NoSQL stores (such as OpenSearch and Redis) 

• Expertise in application, data, and infrastructure architecture disciplines 

• Deep knowledge in Data structures, Algorithms, Machine Learning, Data Mining, Information Retrieval, Statistics. 

• Excellent communication skills and ability to communicate with senior technical and business partners 

 Preferred qualifications, capabilities, and skills

• Expert in at least one of the following areas: Natural Language Processing, Reinforcement Learning, Ranking and Recommendation, or Time Series Analysis. 

• Knowledge of machine learning frameworks: Pytorch, Keras, MXNet, Scikit-Learn 

• Understanding of finance or wealth management businesses is an added advantage 

ABOUT US

JPMorganChase, one of the oldest financial institutions, offers innovative financial solutions to millions of consumers, small businesses and many of the world’s most prominent corporate, institutional and government clients under the J.P. Morgan and Chase brands. Our history spans over 200 years and today we are a leader in investment banking, consumer and small business banking, commercial banking, financial transaction processing and asset management. We recognize that our people are our strength and the diverse talents they bring to our global workforce are directly linked to our success. We are an equal opportunity employer and place a high value on diversity and inclusion at our company. We do not discriminate on the basis of any protected attribute, including race, religion, color, national origin, gender, sexual orientation, gender identity, gender expression, age, marital or veteran status, pregnancy or disability, or any other basis protected under applicable law. We also make reasonable accommodations for applicants’ and employees’ religious practices and beliefs, as well as mental health or physical disability needs. Visit our FAQs for more information about requesting an accommodation.

ABOUT THE TEAM

J.P. Morgan Asset & Wealth Management delivers industry-leading investment management and private banking solutions. Asset Management provides individuals, advisors and institutions with strategies and expertise that span the full spectrum of asset classes through our global network of investment professionals. Wealth Management helps individuals, families and foundations take a more intentional approach to their wealth or finances to better define, focus and realize their goals.

Senior Software Engineer Full stack/ Java track/ UI/ Android

Skills required: -

• Good design skills and exposure to multiple software architectures. -

• Strong understanding of OO/ knowledge of frameworks such as Spring/ Spring Boot/ design and good software development practices. -

• Strong core server-

• side technologies experience (Java) and/or having any of front end/UI skills (Angular, JS, HTML). Knowledge of Android is a huge plus. -

• Strong database skills including schema design. -

• Strong knowhow of data-structures and algorithms. -

• Experience with configuration management, testing, deployment. -

• Hands on development across technologies. -

• Exp- 4-6yrs . -

• Grad- B.E/ B.Tech (Computer Science/ Equivalent) from a premier institute.

Aternately: -

• Strong UI architecture and design skills and keen business sense. -

• Excellent front end/UI skills (JS, HTML, Angular, Flutter) and OR Android skills. -

• Good knowledge of nodeJS. -

• Good knowhow of server-side technologies, exposure to databases. -

• Good exposure to configuration management, testing, deployment. -

• Deep knowhow of data structures, algorithms. -

• Hands on development across technologies. -

• Exp- 4- 6 Yrs

• Grad - B.E/ B.Tech (Computer Science/ Equivalent) from a premier institute.

Responsibilities

·        Design and build advanced applications for the iOS platform.

·        Collaborate with cross-functional teams to define, design, and ship new features.

·        Work on bug fixing and improving application performance.

·        Identify and correct bottlenecks and fix bugs

·        Help maintain code quality, organization, and automatization

·        Support the whole application lifetime from idea through design, testing, release, and maintenance

·        Continuously discover, evaluate, and implement new technologies to maximize development efficiency.

Skills

·        Proficient with Swift and Cocoa Touch

·        Experience with iOS frameworks such as Core Data, Core Animation, etc.

·        Experience with offline storage, threading, and performance tuning

·        Familiarity with RESTful APIs to connect iOS applications to back-end services

·        Understanding of Apple’s design principles and interface guidelines

·        Experience with performance and memory tuning with tools 

·        Familiarity with cloud message APIs and push notifications

·        Knack for benchmarking and optimization

·        Proficient understanding of code versioning tools source tree, git

·        Familiarity with continuous integration

·        Solid grasp of algorithms, memory management, object oriented programming, and MVC Design

Competencies

·        Excellent communication skills

·        Ability to co-ordinate with cross functional teams  

·        Should be a good Team player.

• Align with product managers, business analysts, developers, and other stakeholders to gather product inputs.
• Understand and decipher user journeys while drafting product flows.
• Be able to accept creative feedback and inculcate the same in their writing.
• Highly creative with excellent communication and team player abilities.
• Understand the Transportation industry.
• Research fintech industry snippets (combining online sources, interviews, and studies).
• Prepare well-structured product flows using research, writing techniques, and logic.
• Coordinate with marketing and design teams to illustrate product draft.
• Ensure all-around consistency (style, fonts, images, and tone).

• We are looking for someone with 3-9 years of overall working experience as Content writer.
• Bachelors/Master’s degree in any discipline.
• Ability to meet deadlines and willingness to explore new domains.
• Proven work experience as a Content Writer, Copywriter, or similar role.
• A team player willing to do what it takes.
• Experience doing research using multiple sources.
• Familiarity with product writing and its nuances.
• Able to create audio/video content for product.
• Excellent writing and editing skills in English.
• Ability to communicate technical aspects to people with non-IT background.

• As a Senior Software Test Engineer, your primary responsibility will be to test/qualify new products and enhancements to existing products as part of a cross functional team. You are required to verify and validate software changes for new products, enhancements and defects.
• Responsible for verifying that the software complies with the requirements and that the various customer use-cases can be satisfactorily met by the software.
• Work very closely with the project/baseline manager and development team to understand the architecture and design of the system and the interaction between various software sub-systems and components.
• Prepare and execute test plans for software releases, document and automate test cases to exercise the various features and capabilities of the software, identify and capture bugs and related information in the bug tracking database.
• Develop higher complexity test cases, test suite selection and execute all necessary testing activity required for a customer software release (negative and backward compatibility scenarios) per assignment, group objectives and product requirements.
• Document, enhance and maintain regression test procedures over the lifecycle of the software.
• Execute the software release test plans and certify that a software release has met the quality criterion(s) for various phase exits (pre-alpha, alpha, beta).
• Participate in troubleshooting and triaging of issues with different teams to drive towards root cause identification and resolution
• Independently manage schedule for his/her projects and deliverables, seamlessly works with all key stakeholders to ensure delivery is never on the critical path
• Consistently look out for areas of improvement in all areas of software product development and processes, champions implementation of identified best practices

• Strong understanding of SDLC processes and the QA lifecycle and methodology
• Prior experience in supervisory control systems software would be a plus (but not mandatory)
• Strong knowledge of software test methodologies, tools, standards and processes.
• Good hands on programing experience in any of the scripting/programming languages (C++, C#, Java Script)
• Strong interpersonal skills. Excellent written and verbal communication skills
• Prior experience of working in global teams and interfacing with multiple stake holders

• Design, build and support technical solutions and systems with highest quality.
• Create software architecture, high level design, low level design documents
• Own systems scalability and resilience
• Define and drive adoption of coding best practices
• Define technical roadmap and vision for technical products.
• Work closely with engineering counterparts, product managers and business stakeholders to achieve company goals.
• Coach, mentor and support software development engineers to grow them in their career aspirations. Internal Use—Confidential

Experience & Skills Required

• Strong understanding for technical concepts – Familiarity with microservices, distributed systems and event driven architecture.
• Ability to build scalable systems while thinking strategically balancing short and long term approaches.
• Wide knowledge of architecture standards and design patterns
• Excellent communication skills, fluent English - Communicates effectively in all situations and creates an environment of open and effective communication.
• Strong knowledge of best engineering practices
• Ability to work under minimal guidance
• 7-10 years of software development experience, of which 5+ years should be on salesforce