Rapid7 Logo

Rapid7

Staff AI Engineer

Posted 2 Hours Ago
Be an Early Applicant
Remote or Hybrid
Hiring Remotely in Pune, Maharashtra
Senior level
Remote or Hybrid
Hiring Remotely in Pune, Maharashtra
Senior level
The Staff AI Engineer will collaborate with data scientists and security teams to develop AI-driven solutions, focusing on research, model evaluation, and production deployment in AWS. Responsibilities include mentoring, deploying AI/ML workloads, and enhancing threat detection workflows.
The summary above was generated by AI
Staff AI Engineer - Data Science
Experience: 8-13 years
About the Team
The AI Center of Excellence team includes Data Scientists and AI Engineers that work together to conduct research, build prototypes, design features and build production AI components and systems. Our mission is to leverage the best available technology to protect our customers' attack surfaces. We partner closely with Detection and Response teams, including our MDR service, to leverage AI/ML for enhanced customer security and threat detection. We operate with a creative, iterative approach, building on 20+ years of threat analysis and a growing patent portfolio. We foster a collaborative environment, sharing knowledge, developing internal learning, and encouraging research publication. If you're passionate about AI and want to make a major impact in a fast-paced, innovative environment, this is your opportunity.
The technologies we use include:
  • Python for analysis and modeling, using numpy, pandas, and scientific computing libraries
  • Jupyter notebooks (local & remote)
  • scikit-learn for machine learning
  • Anomaly detection for large-scale unlabeled data analysis
  • LLM/GenAI toolchains: HuggingFace, Transformers, LangChain, CrewAI, Agentic architectures
  • AWS cloud ecosystem, especially SageMaker, Bedrock, Lambda, EKS, S3
  • Agentic AI platforms - multi-agent orchestration, tool use, reasoning frameworks, LLMOps
  • EKS for application deployment
  • Terraform for infrastructure as code

About the Role
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.
This role is ideal for someone who is:
  • deeply strong in data science,
  • hands-on with AWS and ML/LLM infrastructure, and
  • excited about agentic AI, autonomous workflows, tool-augmented LLM systems, and building the future of AI-driven security.

In this role, you will:
  • 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 optimize agentic architectures-tool-calling, decision-making, memory, orchestration, 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

The skills you'll bring include:
Core
  • 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 for classical ML
    • PyTorch/TensorFlow/Keras for deep learning
    • HuggingFace, LangChain, Transformers - especially for LLMs
    • Pandas/NumPy for data preparation
  • Clear intuition for what modeling approaches fit which type of problem
  • Strong communication skills and the ability to present technical concepts to diverse audiences
  • Proven ability to collaborate across engineering, data science, product, and security teams
  • Experience mentoring and guiding junior data scientists

Agentic AI & LLM Expertise (Highly Preferred)
  • 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 working with AWS Bedrock, model selection, latency/cost trade-offs, and LLM deployment patterns

AWS & Cloud (Highly Preferred)
  • Strong experience in AWS ML ecosystem:
    • SageMaker training & inference
    • Bedrock LLM integrations
    • Lambda for serverless workflows
    • EKS for containerized deployments
    • S3, Glue, Athena, IAM best practices
  • Experience with Terraform, IaC, and production deployment workflows

Experience with the following would be advantageous:
  • 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

We know that the best ideas and solutions come from multi-dimensional teams. That's because these teams reflect a variety of backgrounds and professional experiences. If you are excited about this role and feel your experience can make an impact, please don't be shy - apply today.
About Rapid7
At Rapid7, we are on a mission to create a secure digital world for our customers, our industry, and our communities. We do this by embracing tenacity, passion, and collaboration to challenge what's possible and drive extraordinary impact.
Here, we're building a dynamic workplace where everyone can have the career experience of a lifetime. We challenge ourselves to grow to our full potential. We learn from our missteps and celebrate our victories. We come to work every day to push boundaries in cybersecurity and keep our 10,000 global customers ahead of whatever's next.
Join us and bring your unique experiences and perspectives to tackle some of the world's biggest security challenges.
About Rapid7
At Rapid7, our vision is to create a secure digital world for our customers, our industry, and our communities. We do this by harnessing our collective expertise and passion to challenge what's possible and drive extraordinary impact. We're building a dynamic and collaborative workplace where new ideas are welcome.
Protecting 11,000+ customers against bad actors and threats means we're continuing to push the envelope just like we' ve been doing for the past 20 years. If you 're ready to solve some of the toughest challenges in cybersecurity, we're ready to help you take command of your career. Join us.

Top Skills

AWS
Bedrock
Eks
Huggingface
Jupyter Notebooks
Lambda
Langchain
Numpy
Pandas
Python
PyTorch
S3
Sagemaker
Scikit-Learn
TensorFlow
Terraform

Rapid7 India Office

What you need to know about the Pune Tech Scene

Once a far-out concept, AI is now a tangible force reshaping industries and economies worldwide. While its adoption will automate some roles, AI has created more jobs than it has displaced, with an expected 97 million new roles to be created in the coming years. This is especially true in cities like Pune, which is emerging as a hub for companies eager to leverage this technology to develop solutions that simplify and improve lives in sectors such as education, healthcare, finance, e-commerce and more.

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account