Staff AI Engineer Job at Rapid7 (Pune)

Staff AI Engineer

Rapid7

Location:
India , Pune

Category:
IT - Software Development

Contract Type:
Not provided

Salary:

Not provided

Save Job

Apply Position

Job Description:

Rapid7 is seeking a Staff AI Engineer to join our Data Science team as we expand and evolve our growing AI and MLOps efforts. You should have a strong foundation in software engineering and applied R&D in one of the key areas we focus on - traditional machine learning, neural networks, or generative AI. In this intersectional role, you will combine your expertise in AI/ML deployments, cloud systems and software engineering to enhance our product offerings and streamline our platform's functionalities.

Job Responsibility:

Work with security teams to define, scope, and design research efforts for new threat detections, automations, and AI-driven workflows
Collaborate with data scientists to transform research into production-ready solutions, mentoring on both methods and execution
Research, build, and evaluate ML and generative/LLM models, including agentic and multi-step reasoning systems
Design and optimise agentic architectures, including tool-calling, decision-making, memory, orchestration, and evaluation frameworks
Partner with engineering teams to ship AI features, integrating models into high-scale systems
Deploy AI/ML workloads in AWS using SageMaker, Bedrock, Lambda, EKS, Step Functions, and related services
Contribute to our LLMOps and MLOps workflows, improving evaluation pipelines, observability, reproducibility, and governance
Support development of AI security features, improving reasoning, context understanding, automation, and analyst workflows
Embrace agile development, iterative experimentation, and collaborative problem solving
Mentor junior team members and uplift the technical bar for agentic AI and ML engineering

Requirements:

8–12 years of experience as a Data Scientist, ML Engineer, or AI Engineer
Strong end-to-end practical expertise in ML/AI/DS
Able to explore, experiment, and deliver independently
Proficiency in scikit-learn, PyTorch, TensorFlow, Keras, HuggingFace, LangChain, Transformers, Pandas, NumPy
Strong communication skills
Collaboration across engineering, data science, product, and security teams
Experience mentoring and guiding junior data scientists
Agentic AI and LLM expertise
Hands-on experience with agentic AI architectures (CrewAI, LangGraph, LangChain agents, tool-calling, function-chaining)
Experience building multi-step reasoning workflows, autonomous agents, or LLM-powered decision systems
Understanding of prompt engineering, evaluation frameworks, LLM observability, and guardrails
Experience with AWS Bedrock, model selection, latency/cost trade-offs, LLM deployment patterns
AWS and cloud experience
Strong experience in AWS ML ecosystem (SageMaker, Bedrock, Lambda, EKS, S3, Glue, Athena, IAM best practices)
Experience with Terraform, IaC, and production deployment workflows
Experience in the security industry
Deployment of AI and machine learning models
Implementing model risk management strategies (model registries, drift monitoring, hyperparameter tuning)

Nice to have:

Experience in the security industry
Deployment of AI and machine learning models
Implementing model risk management strategies, including model registries, concept/covariate drift monitoring, and hyperparameter tuning

Additional Information:

Job Posted:
May 04, 2026

Employment Type:

Fulltime

Rapid7 - All Job Offers

Job Link Share: