11+ JAX Jobs in Bangalore (Bengaluru) | JAX Job openings in Bangalore (Bengaluru)
Apply to 11+ JAX Jobs in Bangalore (Bengaluru) on CutShort.io. Explore the latest JAX Job opportunities across top companies like Google, Amazon & Adobe.
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
US Market experience mandatory
- Any gender
- Shift timing: 6 PM to 3 AM
- Inhouse desk job
- Excellent communication skills
- Experience: 3 to 8 years
- Individual Contributor role
- B2B SaaS company experience mandatory
- Any graduate
- Interview process: 3 rounds (2 Video Calls + Final Face-to-Face)
- For Account Manager: M365, Azure, or Google Cloud (any one is fine)
5 days of working from the office.
A cab will be arranged for female employees.
Position Title: Account Executive- (SaaS – US Market)
Experience Level: 3-8 Years
Industry: B2B SaaS
Sales Segment: Small & Medium Businesses (SMB)
Market: United States
Employment Type: Full-time
Location: Bangalore, work from the office
Key Responsibilities
Own the end-to-end SMB sales cycle (prospecting to close) for U.S.-based customers
Handle inbound and outbound leads, including cold outreach
Conduct product demos, discovery calls, and negotiations
Build and manage a healthy sales pipeline and accurate forecasts
Consistently achieve or exceed monthly and quarterly quotas
Maintain accurate CRM records and sales activity tracking
Required Qualifications
3–8 years of experience as an Account Executive in B2B SaaS
Strong experience selling to SMB customers (not enterprise-only profiles)
Direct exposure to selling in the U.S. market
Proven track record of quota achievement
Excellent communication, negotiation, and closing skills
Hands-on experience with CRM tools such as HubSpot
Strong understanding of U.S. buyer behavior and SMB needs
Preferred Qualifications
Experience in fast-paced SaaS startups or growth-stage companies
Familiarity with high-volume deal environments ($5k–$100k ARR)
Strong consultative selling approach
Strong understanding of U.S. buyer behavior and SMB needs
Experience in SAP SAC
Experience in end to end implementations
Experience in Configuring SAC to connect and integrate with SAP BW, SAP HANA, and other data sources
Experience in Design and optimize data models in SAP BW, ensuring data is structured efficiently for reporting and analytics
Experience in Implementing data extraction, transformation, and loading (ETL) processes from various sources into SAP BW.
Experience in data modeling, data integration, and reporting using SAP BW and SAP HANA
Experience in Financial Consolidation and Group Reporting.
Experience in Documentation
Experience in Understanding of report building, setting up dimensions and hierarchy, ETL and UAT
Role Overview
We’re looking for an Elixir Developer who is passionate about building scalable, high performance backend systems. You’ll work closely with our engineering team to design, develop, and maintain reliable applications that power mission-critical systems.
Key Responsibilities
• Develop and maintain backend services using Elixir and Phoenix framework.
• Build scalable, fault-tolerant, and distributed systems.
• Integrate APIs, databases, and message queues for real-time applications.
• Optimize system performance and ensure low latency and high throughput.
• Collaborate with frontend, DevOps, and product teams to deliver seamless solutions.
• Write clean, maintainable, and testable code with proper documentation.
• Participate in code reviews, architectural discussions, and deployment automation
Must-Have Qualifications
• 2–4 years of hands-on experience in Elixir (or strong functional programming background).
• Experience with Phoenix, Ecto, and RESTful API development.
• Solid understanding of OTP (Open Telecom Platform) concepts like GenServer, Supervisors, etc.
• Proficiency in PostgreSQL, Redis, or similar databases.
• Familiarity with Docker, Kubernetes, or cloud platforms (AWS/GCP/Azure).
• Understanding of CI/CD pipelines, version control (Git), and agile development.
Good-to-Have Skills
• Experience with microservices architecture or real-time data systems.
• Knowledge of GraphQL, LiveView, or PubSub.
• Exposure to performance profiling, observability, or monitoring tools.
Job Role - Shopify Developer
Responsibilities
Create an engaging and knowledgeable online Shopify store to increase sales and revenue
Create bespoke Shopify themes and change pre-existing templates per pre-defined brand
guidelines
Be an expert in all aspects of the ECommerce platform
Work with the UX and UI Design Teams to create unique, strong, and inventive front-end user
experiences
Ensure a smooth connection with Marketing Tools, Platform APIs, and Shopify Apps to optimise
the shop for overall efficiency and functionality
Test and debug websites regularly to improve performance
Based on their findings, they provide technical support and coordination, protecting enterprises
against failure
Increase conversion rates by optimising website
Requirements & Skills
A bachelor's degree in information technology, computer science, or a related discipline
Broad shop construction, launch, and maintenance expertise, as well as a general understanding
of the admin system
Proven skills in HTML5, CSS3, and JavaScript, as well as a thorough grasp of the DOM
Previous experience working with a custom theme and/or the Storefront API
Working knowledge of Shopify's theming system and Liquid templating
Previous experience implementing/debugging third-party Shopify apps, as well as building
unique solutions if needed
In-depth knowledge and expertise with vanilla JavaScript, jQuery, ES2015/ES6, and current
JavaScript frameworks
Working knowledge of Shopify's object/properties, AJAX API, and Meta fields
Extensive testing and debugging abilities using the browser console and other tools
Prior experience developing responsive layouts for desktop, tablet, and mobile devices
Working knowledge of third-party services and APIs, as well as Shopify plugins
Capability to communicate effectively and provide proactive feedback
Role: Software Development - Other
Industry Type: Advertising & Marketing
Department: Engineering - Software & QA
Employment Type: Full Time, Permanent
Role Category: Software Development
Education
UG: Any Graduate
Key Skills
Skills highlighted with ‘‘ are preferred keyskills
Html/Css
Performance OptimizationApp DevelopmentDebuggingJavascriptE-commerceliquid tempelating languageVersion ControlTroubleshootingResponsive DesignSEO best practices
Skills: Java, spring, hibernate, AWS(mandatory), microservices architecture, redis(good to have)
Team Size: Current team size of tech team is 65+
Interview Process: 4 rounds( 1st round - Problem Solving/Data Structures, 2nd Round- Coding, 3rd Round- Technical Discussion and Final round with the HR).
Target Companies:- B2C fintech companies, app based product companies, edtech companies
Location: Bangalore(Currently WFH but later on will be following a hybrid model)
Roles & Responsibilities
Job Description: (Mandate Skills)
· Exceptional architecture and design skills using Object Oriented Programming concepts and Design patterns
· Strong problem solving and analytical skills
· Hands on Java Engineer, with experience building consumer facing or enterprise applications using Java stack
· Ability to manage small teams (4-6 members) using agile development process
Do You Know? (Skills good to have)
· Exposure to building service oriented distributed systems
· In building systems that process big data in a distributed environment, either in real time streaming or offline batching.
· In messaging systems like Kafka, RabbitMQ, kinesis, etc.
· In real time computation tools like Storm / Spark or Hadoop based tools.
· In Data warehousing technologies like Redshift, BigQuery, etc.
What Can We Offer You:
We offer a great working environment - Very collaborative. You will get plenty of learning opportunities and challenging problems to work on. You’ll go to sleep everyday knowing that you’ve made a difference in people’s lives. Your work will be reaching out to millions of users.
Deep-Rooted.Co is on a mission to get Fresh, Clean, Community (Local farmer) produce from harvest to reach your home with a promise of quality first! Our values are rooted in trust, convenience, and dependability, with a bunch of learning & fun thrown in.
Founded out of Bangalore by Arvind, Avinash, Guru and Santosh, with the support of our Investors Accel, Omnivore & Mayfield, we raised $7.5 million in Seed, Series A and Debt funding till date from investors include ACCEL, Omnivore, Mayfield among others. Our brand Deep-Rooted.Co which was launched in August 2020 was the first of its kind as India’s Fruits & Vegetables (F&V) which is present in Bangalore & Hyderabad and on a journey of expansion to newer cities which will be managed seamlessly through Tech platform that has been designed and built to transform the Agri-Tech sector.
Deep-Rooted.Co is committed to building a diverse and inclusive workplace and is an equal-opportunity employer.
How is this possible? It’s because we work with smart people. We are looking for Engineers in Bangalore to work with thehttps://www.linkedin.com/in/gururajsrao/"> Product Leader (Founder) andhttps://www.linkedin.com/in/sriki77/"> CTO and this is a meaningful project for us and we are sure you will love the project as it touches everyday life and is fun. This will be a virtual consultation.
We want to start the conversation about the project we have for you, but before that, we want to connect with you to know what’s on your mind. Do drop a note sharing your mobile number and letting us know when we can catch up.
Purpose of the role:
* As a startup we have data distributed all across various sources like Excel, Google Sheets, Databases etc. We need swift decision making based a on a lot of data that exists as we grow. You help us bring together all this data and put it in a data model that can be used in business decision making. * Handle nuances of Excel and Google Sheets API. * Pull data in and manage it growth, freshness and correctness. * Transform data in a format that aids easy decision-making for Product, Marketing and Business Heads. * Understand the business problem, solve the same using the technology and take it to production - no hand offs - full path to production is yours.
Technical expertise:
* Good Knowledge And Experience with Programming languages - Java, SQL,Python. * Good Knowledge of Data Warehousing, Data Architecture. * Experience with Data Transformations and ETL; * Experience with API tools and more closed systems like Excel, Google Sheets etc. * Experience AWS Cloud Platform and Lambda * Experience with distributed data processing tools. * Experiences with container-based deployments on cloud.
Skills:
Java, SQL, Python, Data Build Tool, Lambda, HTTP, Rest API, Extract Transform Load.

The company is a global leader in digital mental health(WY1)
- Build beautiful, user-friendly, performant, and effective new features
- Take ownership of the company’s iOS platform
- Continuously improve stability and performance of the company’s iOS platform
- Be the user’s voice and help make product decisions
- Deliver beautiful delightful user experiences
- Exhibits knowledge and behavior to adopt Information Security and Privacy practices at their work.
- You think iOS, design for iOS, build for iOS! Technical
- At least 1.5 - 3 years experience on native iOS platform using the Cocoa framework
- Proficiency with Swift and familiarity with Objective-C
- Experience with RxSwift or Combine framework is preferred
- Strong knowledge of iOS Frameworks like Core data, CoreAnimation, etc
- Expertise with Auto-layouts for UI development
- Experience building a modular code base in MVVM Team Building & Leadership
- Want to interact with all levels of the company including senior leaders and cross-functional teams
- Love problem solving and thrive on handling multiple projects and tasks
- Foster strong professional relationships with the full organization
CasaOne - CasaOne is a home furniture rental company that helps property managers, realtors, stagers and residents experience and access furniture, furnishings, decor, and equipment on demand. Our full-service offering includes a wide selection of products, affordable monthly rental plans, 3D visualizations, rent-to-own options and white-glove service in as early as 2-3 business days. www.casaone.com
BureauOne - BureauOne converts furniture into a utility for businesses. BureauOne is a furniture rental company that helps businesses experience and access furniture, fixtures, and equipment without going through the myriad challenges that are associated with buying office furniture. BureauOne offers a full-service product including white-glove storage, delivery, and installation on-site. www.bureauone.com.
We are backed by Accel Partners & other prominent investors and have offices in the SF Bay Area (HQ), LA, DC, NY, NJ,Chicago, Philadelphia, San Diego, Connecticut & Bangalore.
Our teams consist of high performing individuals who enable each other to bring their best to every concept, resulting in brands renowned in their markets for perfect insight, impeccable service & the constant ability to surprise with excellence. We onboard experts who inspire us in the way they design solutions, craft brands or simply add sense to our products & services.
We believe technology is only a tool for great engineering minds. Your understanding of core principles is of greater value to us than the tools used.
At CasaOne and BureauOne you will have an opportunity:
• to increase your rate of learning exponentially by defining hard problems and solving them
• to partake in a high-growth journey and increase revenues 5x+ Y-o-Y
• to be an early innovator in the shifting trend: ‘ownership economy’ -> ‘access economy’
• to build a category-defining platform for FF&E (Furniture, Fixture, and Equipment) leasing
• to build high-performance teams.
What you will work on:
Build a scalable API platform that will enhance our customer experience & propel our logistics. You will be part of our Bangalore
team of ambitious and talented engineers, who put their best together to build architecturally sound & scalable systems.
Requirements and Qualifications
• Bachelor’s or Master’s degree in engineering
• Good understanding of algorithms, data structures & design patterns
• A minimum of 4 years of work experience
• Building distributed systems & service-oriented architecture
• Asynchronous programming, Test Driven Development (TDD)
• Writing (delightful) APIs & integration patterns
• RDBMS & NoSql databases
• Continuous integration & deployment (CI/CD) tools like git, Jenkins
• Cloud computing platforms - AWS/ Azure/ Google Cloud
Good to know
CasaOne backend services are written in NodeJS. Experience in NodeJS is good to have, not a must. We believe technology is only a tool for great engineers. We value your understanding of core principles more than the tools used.
What happens next
You read this, you say ‘ah, that’s me, that’s what I would love to do in the near future!. Submit your resume and someone from our team will reach out to you.
CasaOne is an equal opportunity employer (EEO) & we value diversity in our teams.







