2+ JAX Jobs in Hyderabad | JAX Job openings in Hyderabad

Minimum qualifications:

• Bachelor’s degree or equivalent practical experience.
• 5 years of experience with software development in one or more programming languages.
• 3 years of experience testing, maintaining, or launching software products, and 1 year of experience with software design and architecture.
• 3 years of experience with one or more of the following: Speech/audio (e.g., technology duplicating and responding to the human voice), reinforcement learning (e.g., sequential decision making), ML infrastructure, or specialization in another ML field.
• 3 years of experience with ML infrastructure (e.g., model deployment, model evaluation, optimization, data processing, debugging).

Preferred qualifications:

• Master’s degree or PhD in Computer Science or a related field with a focus on Artificial Intelligence (AI) or Machine Learning (ML).
• 7 years of software experience designing and deploying large-scale AI/ML applications and complex models.
• Experience with MLOps, establishing best practices for production model monitoring, maintenance, and lifecycle management.
• Experience leading and mentoring engineers, and translating complex business objectives into scalable AI-driven roadmaps.
• Experience architecting scalable data pipelines using BigQuery, Beam, or Dataflow, and deploying models on Google Cloud Platform.
• Understanding of Python and frameworks like TensorFlow, JAX, and Scikit-learn, with deep statistical modeling knowledge.

About the job

The People Operations Engineering team is dedicated to building innovative and scalable technology solutions that empower a global workforce. Leveraging the power of data, AI, and machine learning, the team optimizes HR processes, enhances employee experiences, and drives organizational efficiency. Joining this team offers the opportunity to lead the application of AI to solve complex, real-world business challenges.

We are seeking a highly experienced and results-oriented Senior AI Software Engineer to join the Human Resources Engineering (HRE) team in Hyderabad.

In this role, you will serve as a technical leader, driving the design, development, and deployment of sophisticated AI-powered solutions that significantly impact employees across the organization. You will work on large-scale infrastructure, mentor a talented team of engineers, collaborate with cross-functional stakeholders, and help shape the future of HR through innovative AI applications. You will be responsible for translating complex business challenges and strategic objectives into scalable, efficient AI/ML systems.

As part of the Corporate Engineering organization, you will build world-class business solutions that support a global enterprise. The team delivers end-to-end platforms, tools, and experiences that enable employees to work more effectively and drive innovation across the organization.

Responsibilities

• Lead the technical design and architecture of complex, scalable AI/ML systems and infrastructure within the HR engineering domain, ensuring high reliability, performance, and long-term sustainability.
• Drive the development of advanced AI/ML algorithms for HR applications such as talent acquisition and employee engagement using Python and frameworks like TensorFlow and JAX.
• Architect large-scale data pipelines for data ingestion, cleaning, and feature engineering using modern data processing frameworks (e.g., Beam, Dataflow), while leading root cause analysis and system troubleshooting.
• Own the end-to-end MLOps lifecycle, including model deployment, monitoring, optimization, and maintenance, while establishing best practices for software development and code quality.
• Collaborate with cross-functional stakeholders to translate business needs into scalable AI/ML solutions, mentor engineers, and drive the adoption of emerging AI technologies across the organization.

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