CrawlJobs Logo
Rapid7 Logo Rapid7 · IT - Software Development

Staff AI Engineer

India, Pune · Job Posted May 04, 2026
Apply Position
Job Link Share

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
Rapid7 - All Job Offers

Looking for more opportunities?

Search for other job offers that match your skills and interests.

Suggested Tags: Frontend Developer Secretary Talent Manager Salesforce Administrator Data Analyst
Similar Jobs for

Staff AI Engineer

8 matching positions

Staff Ai Engineer

As a Staff AI Engineer, you will play a key role in designing and delivering hig...
Location
Location
India , Bengaluru
Salary
Salary:
Not provided
teradata.com Logo
Teradata
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Bachelor's degree in Computer Science or equivalent from a recognized institution
  • 8+ years of experience in backend services, distributed systems, or data platform development
  • Strong proficiency in Java, Go, or Python for service development
  • Deep understanding of design principles, distributed system patterns, and service architecture
  • Hands-on experience designing and developing RESTful APIs
  • Experience with SQL and NoSQL databases and data modelling
  • Strong debugging, problem solving, and troubleshooting skills
  • Experience with modern containerization and orchestration tools such as Kubernetes
  • Knowledge of public cloud platforms
  • Experience with AI productivity tools (e.g., GitHub Copilot)
Job Responsibility
Job Responsibility
  • Design, architect, develop, and maintain high quality systems, services, and applications with an emphasis on scalability, reliability, and performance
  • Collaborate with cross-functional engineers and product partners to shape architecture and consistently deliver end to end features
  • Build and integrate robust RESTful APIs, ensuring security, data consistency, and maintainability
  • Work with SQL and NoSQL databases to implement efficient data models and service access patterns
  • Apply and experiment with AI/ML technologies, including agentic AI and large language models (LLMs)
  • Use AI powered engineering tools to improve development quality, speed, and productivity
  • Mentor engineers, supporting them in technical planning, implementation, and best practices
  • Identify and resolve system performance bottlenecks, optimizing code, architecture, and infrastructure
  • Write unit and integration tests and participate in code reviews to uphold engineering excellence
  • Investigate production issues, ensuring timely and effective solutions
  • Fulltime
Read More
Arrow Right

Staff AI Engineer

As a Staff AI Engineer, you will be a senior technical leader responsible for de...
Location
Location
India , Hyderabad
Salary
Salary:
Not provided
teradata.com Logo
Teradata
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Bachelor’s degree in Computer Science, Engineering, or equivalent practical experience
  • 8+ years of experience building backend services, distributed systems, or data/AI platforms
  • Strong proficiency in Java, Go, or Python, with experience building large‑scale services
  • Deep understanding of distributed system design, scalability, fault tolerance, and cloud‑native architectures
  • Proven experience designing and operating production systems with SQL and NoSQL data stores
Job Responsibility
Job Responsibility
  • Lead the design and evolution of large‑scale, distributed AI systems that power Teradata’s AI platform and AI‑native products
  • Own end‑to‑end architecture for critical AI capabilities such as agentic workflows, RAG pipelines, vector search, semantic retrieval, and AI orchestration frameworks
  • Drive technical strategy and architectural consistency across multiple engineering teams
  • Design and implement production‑grade AI systems using LLMs, embeddings, vector databases, and agent‑based architectures
  • Build scalable, secure, and reusable platform services and APIs supporting AI workloads across the software development lifecycle
  • Define and implement guardrails for reliability, safety, governance, and cost control in enterprise AI systems
  • Partner with product management, architecture, research, and cloud platform teams to translate business requirements into scalable AI solutions
  • Influence roadmap decisions by providing deep technical insight, trade‑off analysis, and long‑term platform thinking
  • Act as a technical escalation point for complex system design, performance, and reliability challenges
  • Drive best practices for testing, observability, evaluation, and production readiness of AI systems
What we offer
What we offer
  • People-first culture
  • Flexible work model
  • Focus on well-being
  • Inclusive environment
  • Fulltime
Read More
Arrow Right

Staff AI Engineer

We are looking for a visionary and technically deep AI Architect to lead the des...
Location
Location
India , Pune, Maharashtra; Karnataka
Salary
Salary:
Not provided
teradata.com Logo
Teradata
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • 10+ years of experience in software engineering or technology architecture
  • At least 4 years focused on AI, ML, and Generative AI solutions at enterprise scale
  • Deep hands-on expertise with Large Language Models (LLMs) including model selection, prompt engineering, fine-tuning (LoRA, PEFT), and evaluation techniques
  • Proven experience designing and implementing MCP servers and structured context management strategies for LLM applications
  • Strong working knowledge of Generative AI frameworks and orchestration tools — LangChain, LlamaIndex, LangGraph, AutoGen, CrewAI, or equivalent agentic platforms
  • Hands-on experience with RAG architectures, vector databases (Pinecone, Weaviate etc), and embedding models
  • Proficiency in Python-based AI/ML development, including libraries such as HuggingFace Transformers, PyTorch, TensorFlow, scikit-learn, and OpenAI SDK
  • Experience with AI cloud platforms and services — Azure OpenAI, AWS Bedrock, Google Vertex AI, or equivalent enterprise AI infrastructure
  • Solid understanding of MLOps practices including model versioning, experiment tracking (MLflow, Weights & Biases), CI/CD for ML, and model monitoring in production
  • Familiarity with AI agent design patterns, multi-agent systems, tool-use frameworks, and autonomous workflow orchestration
Job Responsibility
Job Responsibility
  • Define and own the enterprise AI architecture strategy, establishing standards, reference architectures, and technology roadmaps for AI/ML adoption
  • Architect and oversee the design of LLM-powered applications including RAG pipelines, AI agents, and agentic workflow systems
  • Design and govern MCP server implementations, enabling structured, context-aware interactions between LLMs and enterprise data sources
  • Lead the evaluation and selection of AI/ML platforms, LLM providers (OpenAI, Anthropic, Azure OpenAI, Google Gemini, open-source models), and supporting infrastructure
  • Architect prompt engineering frameworks, fine-tuning pipelines, and model evaluation strategies to ensure LLM output quality, accuracy, and reliability
  • Design AI orchestration layers using frameworks such as LangChain, LlamaIndex, AutoGen, CrewAI, or custom agentic architectures for multi-step reasoning workflows
  • Establish AI data pipelines for model training, embeddings generation, vector database management, and real-time inference serving
  • Define and enforce responsible AI practices including bias detection, explainability, hallucination mitigation, content safety guardrails, and regulatory compliance
  • Collaborate with engineering leads, data scientists, and product teams to embed AI capabilities into existing platforms and new product initiatives
  • Mentor and upskill engineering teams on AI/ML technologies, architectural patterns, and emerging developments in the AI ecosystem
What we offer
What we offer
  • People-first culture
  • Flexible work model
  • Focus on well-being
  • Inclusive environment
  • Fulltime
Read More
Arrow Right

Staff AI Engineer

The Staff Software Engineer will be responsible for building, maintaining, monit...
Location
Location
United States , Austin; San Francisco; San Mateo; Roseville; Irvine
Salary
Salary:
173000.00 - 200000.00 USD / Year
goodleap.com Logo
GoodLeap
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Bachelor's degree
  • 7+ years experience in backend software, ML, or AI engineering
  • Highly proficient in ML and AI engineering tools, python/FastAPI, A2A protocol, agent design and evaluation, prompt and context engineering
  • Adept with agile software development lifecycle and DevOps principles
  • Passion for software development, emerging technologies and culture of innovation
  • Excellent communication and interpersonal skills
Job Responsibility
Job Responsibility
  • Collaborate with product, architecture, and design leads to deliver highly-available, fault-tolerant products and services
  • Work on significant and unique technical challenges, evaluate and recommend solutions, and guide decision making by considering technical tradeoffs
  • Grasp both the technical and business perspective so you can help drive innovation
  • Work autonomously and be self-disciplined, requiring no supervision or guidance
  • Mentor and coach the team members to grow both their technical skills and soft skills
What we offer
What we offer
  • May be eligible for a bonus and equity
  • Fulltime
Read More
Arrow Right

Staff AI Engineer

As the staff engineer in our new Generative AI department, you will define the t...
Location
Location
Singapore , Singapore
Salary
Salary:
Not provided
airwallex.com Logo
Airwallex
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • 10+ years in backend software development, with a passion for AI and language technologies
  • Proven track record of delivering major impact
  • Proficiency in Python, Typescript or Javascript
  • Experience in design and development of large-scale distributed, high concurrency, high load, high availability systems
  • Experience in vector databases
  • A relevant degree in Computer Science, Mathematics or related fields
Job Responsibility
Job Responsibility
  • Spearhead the creation of AI solutions to enable growth and drive internal efficiency across the business
  • Design and implement robust API and system architecture for new AI applications
  • Assist in AI application roadmapping and strategy
  • Collaborate effectively with a diverse, cross-functional team to leverage existing AI technologies, and research and implement new ones, to meet product goals
  • Providing prompt and effective solutions or workarounds in response to application issues or software bugs, maintaining high client satisfaction
  • Embrace the challenges of a fast-paced, high-growth environment with a focus on innovation
  • Fulltime
Read More
Arrow Right

Staff AI Engineer

As we work towards building out the Context Layer for the Agentic Enterprise, we...
Location
Location
United States , Palo Alto
Salary
Salary:
Not provided
workato.com Logo
Workato
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Bachelor's/Master's/PhD degree in Statistics, Mathematics, Computer Science, or another quantitative field
  • 7+ years of backend engineering experience with 3+ years in search, information retrieval, or related fields
  • Strong proficiency in Python
  • Hands-on experience with search engines (Opensearch or Elasticsearch)
  • Strong understanding of information retrieval concepts spanning traditional methods (TF-IDF, BM25) and modern neural search techniques (vector embeddings, transformer models)
  • Experience with text processing, NLP, and relevance tuning
  • Experience with relevance evaluation metrics (NDCG, MRR, MAP)
  • Experience with large-scale distributed systems
  • Proficiency in Knowledge Graph construction and optimization is a plus
  • Strong analytical and problem-solving skills
Job Responsibility
Job Responsibility
  • Lead the development of advanced query understanding systems that parse natural language, resolve ambiguity, and infer user intent
  • Design and deploy learning-to-rank models that optimize relevance using behavioral signals, embeddings, and structured feedback
  • Build and scale robust Entity Recognition pipelines that enhance document understanding, enable contextual disambiguation, and support entity-aware retrieval
  • Architect next-gen search infrastructure capable of supporting highly dynamic document corpora and real-time indexing
  • Create and maintain graph-based knowledge systems that enhance LLM capabilities through structured relationship data
  • Drive improvements in query rewriting, intent classification, and semantic search, using both statistical and neural methods
  • Own the design of evaluation frameworks for offline/online relevance testing, A/B experimentation, and continual model tuning
  • Collaborate with product and applied research teams to translate user needs into data-informed search innovations
  • Produce clean, scalable code and influence system architecture and roadmap across the relevance and platform stack
What we offer
What we offer
  • vibrant and dynamic work environment
  • multitude of benefits they can enjoy inside and outside of their work lives
Read More
Arrow Right

Staff AI Engineer

We're seeking a Staff AI Engineer to join our growing team and help build the ne...
Location
Location
Canada
Salary
Salary:
182000.00 - 220000.00 CAD / Year
procurify.com Logo
Procurify
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • 10+ years of software engineering experience with a significant portion of that focussed on AI/ML, building and deploying AI features in production environments
  • Deep understanding of LLMs, prompt engineering, RAG architectures, vector databases (Databricks/pgVector), and agentic AI frameworks (LangChain, Google ADK, or similar)
  • Strong Python expertise with modern frameworks (FastAPI, Flask, Django) and microservice architecture patterns
  • Experience with AWS/GCP/Azure, containerization (Docker/Kubernetes), and CI/CD pipelines
  • Strong foundation in statistical methods, data pipelines, and ETL processes
  • Experience with A/B testing, evaluation frameworks, and hypothesis-driven development to validate AI features
  • Deep commitment to code quality, testing, monitoring, and observability in AI systems
  • Comfortable navigating ambiguity and rapidly adopting new AI technologies and methodologies
  • Strong communication skills with ability to explain technical concepts to both technical and non-technical stakeholders
Job Responsibility
Job Responsibility
  • Build AI-Powered Features: Design, develop, and deploy production-grade AI applications that solve real customer problems, from concept to deployment
  • Architect Scalable Systems: Create robust backend architectures that support AI workloads, ensuring low latency, high reliability, and seamless integration with existing services
  • Drive AI Innovation: Implement and optimize agentic AI systems, RAG pipelines, and multi-agent workflows using modern LLM frameworks
  • Own the Full Stack: Take end-to-end ownership of features, from database design and API development to AI model integration and monitoring
  • Collaborate Cross-Functionally: Work closely with product, design, and data teams to translate business requirements into technical solutions
  • Optimize and Iterate: Run experiments, A/B tests, and evaluations to continuously improve AI system performance and user experience
What we offer
What we offer
  • Flexible working: remote-first organization with flexible working hours
  • Responsible time off
  • Extended health benefits: competitive health, vision, and dental package along with an Employee Assistance Program (EAP), and a health and wellness spending account
  • Community initiatives: regular community days, Donate Your Day program, lunch and learns
  • Stock options
  • DEI initiatives: diversity, equity, and inclusion roundtable
  • Fulltime
Read More
Arrow Right

Staff AI Engineer

You will operate as a Staff+ AI scientist/engineer on the Orchestration team. Yo...
Location
Location
Australia , Sydney
Salary
Salary:
Not provided
heidihealth.com Logo
Heidi
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Staff‑level track record shipping search, NLP, or LLM systems that serve real users at scale
  • Mastery of Python and SQL
  • Strong software engineering fundamentals, testing strategy, and API/service design
  • Depth in modern IR/NLP: embeddings, ANN indexes, re‑rankers, retrieval‑augmented generation, and prompt/program synthesis
  • Experience building data pipelines: parsing PDFs/HTML, OCR when needed, metadata extraction, and content hashing/versioning
  • Familiarity with PyTorch, plus distributed training/inference patterns
  • MLOps and reliability: containers, Kubernetes, feature/model registries, experiment tracking, monitoring, and alerting
  • Evidence of rigorous evaluation design: offline metrics, human‑in‑the‑loop judging, power analysis for online tests
  • Clear thinking on safety: hallucination controls, calibration, abstention, red‑teaming, and privacy/security by design
  • Ability to lead cross‑functional initiatives and make crisp decisions with incomplete information
Job Responsibility
Job Responsibility
  • Define the end‑to‑end architecture for literature and guideline ingestion, normalization, metadata extraction, de‑duplication, and versioning
  • Build hybrid search and retrieval: lexical + vector + re‑ranking, with tight latency budgets and cost controls
  • Design grounding and answer synthesis that cite sources, preserve provenance, and expose confidence and abstention
  • Lead model work across prompting, fine‑tuning, distillation, and tool use to improve faithfulness, coverage, and utility
  • Stand up gold‑standard evaluation: offline IR metrics (nDCG, MAP, recall), factuality/faithfulness audits, and human review with adjudication
  • Run online experiments at scale. Define guardrails, KPIs, and ship A/Bs to measure impact on clinician workflows
  • Productionize services with observability, tracing, canaries, rollbacks, and incident playbooks
  • Set data governance for medical content: access control, PHI handling, audit logs, and retention policies
  • Partner with clinicians to define intents, schemas, and acceptance criteria. Convert ambiguous questions into testable specs
  • Coach engineers and scientists. Raise the technical bar through design docs, reviews, and reusable components
What we offer
What we offer
  • Flexible hybrid working environment, with 3 days in the office
  • Additional paid day off for your birthday and wellness days
  • Special corporate rates at Anytime Fitness in Melbourne, Sydney tbc
  • A generous personal development budget of $500 per annum
  • Learn from some of the best engineers and creatives, joining a diverse team
  • Become an owner, with shares (equity) in the company, if Heidi wins, we all win
  • The rare chance to create a global impact as you immerse yourself in one of Australia’s leading healthtech startups
  • If you have an impact quickly, the opportunity to fast track your startup career
  • Fulltime
Read More
Arrow Right