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

Job Title : Java Full Stack Developer

Location : Bangalore (Work From Office)

Experience : 4+ Years

Employment Type : Contractual

Job Description :

We are seeking a highly skilled Java Full Stack Developer to join our dynamic team in Bangalore.

The ideal candidate will have strong expertise in full stack development using Java, Spring Boot, and Angular, with experience working in Agile development environments.

Prior experience in the Healthcare Imaging domain is highly desirable.

Key Responsibilities :

• Design, develop, and maintain scalable web applications using Java (Spring Boot) and Angular.
• Develop and maintain RESTful microservices in a cloud-based environment.
• Collaborate with cross-functional teams to deliver end-to-end solutions in an Agile development framework.
• Ensure high performance, security, and responsiveness of applications.
• Participate in code reviews, unit testing, and system integration.
• Troubleshoot and debug applications as required.

Required Skills & Qualifications :

• Strong proficiency in Java and Spring Boot framework.
• Hands-on experience with Angular or similar frontend frameworks.
• Experience with Microservice Architecture and REST APIs.
• Working knowledge of Agile development methodologies (Scrum, Kanban).
• Experience with version control systems like Git.
• Familiarity with CI/CD pipelines and DevOps tools is a plus.
• Healthcare Imaging domain experience is a strong advantage.
• Excellent problem-solving and communication skills.

Nice to Have :

• Knowledge of cloud platforms such as AWS or Azure.
• Exposure to containerization tools like Docker and Kubernetes.
• Understanding of healthcare data formats like DICOM, HL7, etc.

Job Title: DevOps - 3

Roles and Responsibilities:

• Develop deep understanding of the end-to-end configurations, dependencies, customer requirements, and overall characteristics of the production services as the accountable owner for overall service operations
• Implementing best practices, challenging the status quo, and tab on industry and technical trends, changes, and developments to ensure the team is always striving for best-in-class work
• Lead incident response efforts, working closely with cross-functional teams to resolve issues quickly and minimize downtime. Implement effective incident management processes and post-incident reviews
• Participate in on-call rotation responsibilities, ensuring timely identification and resolution of infrastructure issues
• Possess expertise in designing and implementing capacity plans, accurately estimating costs and efforts for infrastructure needs.
• Systems and Infrastructure maintenance and ownership for production environments, with a continued focus on improving efficiencies, availability, and supportability through automation and well defined runbooks
• Provide mentorship and guidance to a team of DevOps engineers, fostering a collaborative and high-performing work environment. Mentor team members in best practices, technologies, and methodologies.
• Design for Reliability - Architect & implement solutions that keeps Infrastructure running with Always On availability and ensures high uptime SLA for the Infrastructure
• Manage individual project priorities, deadlines, and deliverables related to your technical expertise and assigned domains
• Collaborate with Product & Information Security teams to ensure the integrity and security of Infrastructure and applications. Implement security best practices and compliance standards.

Must Haves

• 5-8 years of experience as Devops / SRE / Platform Engineer.
• Strong expertise in automating Infrastructure provisioning and configuration using tools like Ansible, Packer, Terraform, Docker, Helm Charts etc.
• Strong skills in network services such as DNS, TLS/SSL, HTTP, etc
• Expertise in managing large-scale cloud infrastructure (preferably AWS and Oracle)
• Expertise in managing production grade Kubernetes clusters
• Experience in scripting using programming languages like Bash, Python, etc.
• Expertise in skill sets for centralized logging systems, metrics, and tooling frameworks such as ELK, Prometheus/VictoriaMetrics, and Grafana etc.
• Experience in Managing and building High scale API Gateway, Service Mesh, etc
• Systematic problem-solving approach, coupled with strong communication skills and a sense of ownership and drive
• Have a working knowledge of a backend programming language
• Deep knowledge & experience with Unix / Linux operating systems internals (Eg. filesystems, user management, etc)
• A working knowledge and deep understanding of cloud security concepts
• Proven track record of driving results and delivering high-quality solutions in a fast-paced environment
• Demonstrated ability to communicate clearly with both technical and non-technical project stakeholders, with the ability to work effectively in a cross-functional team environment. 

1. Generate qualified opportunities for the company by rigorously prospecting and researching in the given region/ market/ industry. Information about MEDDIC is a plus
2. Being the face of CM.com’s and introducing the company and the product to the prospects so as to generate funnel.
3. Write creative communication (emails/ Inmails) to grab  the attention of prospects.
4. Ideate and innovate methods to improve SDR metrics

• Compile and analyze data, processes, and codes to troubleshoot problems and identify areas for improvement.
• Collaborating with the front-end developers and other team members to establish objectives and design more functional, cohesive codes to enhance the user experience.
• Developing ideas for new programs, products, or features by monitoring industry developments and trends.
• Recording data and reporting it to proper parties, such as clients or leadership.
• Participating in continuing education and training to remain current on best practices, learn new programming languages, and better assist other team members.
• Taking lead on projects, as needed.