Software Engineer II - Python AI Engineer

Rocket Close, LLC

Detroit (MI)

On-site

USD 140,000 - 190,000

Full time

6 days ago
Be an early applicant

A complete application in a minute

Tailored resume and cover letter, ready to send.

Get past ATS filters

Job summary

Rocket Close seeks a senior AI/ML engineer to design autonomous agents and orchestrate complex workflows using AgentCore and LangGraph. You will productionize ML models in AWS, collaborate with Data Scientists to move research into scalable services, and drive observability with Splunk and CloudWatch.

You will explore latest AI frameworks, implement event-driven architectures, and contribute to a high-velocity engineering culture within a modern cloud environment.

Qualifications

  • Master's degree in CS/ML or equivalent experience (3-5+ years in production engineering).
  • Expert-level Python for distributed, scalable cloud-native services.
  • Production ML/DS environment experience bridging research code to production software.

Responsibilities

  • Design and implement autonomous agents using AgentCore and orchestrate via LangGraph for visibility and manageability.
  • Productionize ML models in AWS environments with collaboration from Data Scientists.
  • Research and adopt new AI tools, frameworks, and papers to improve velocity and service quality.
  • Build and maintain cloud-native infrastructure for AI/ML inference and agent execution (AWS).
  • Implement deep monitoring and alerting for all services (LangGraph, Splunk, CloudWatch).
  • Maintain code quality through rigorous reviews and define engineering standards for DS teams.
  • Collaborate with teams to derive solutions amid evolving business needs.

Skills

Python expert
Distributed systems
Cloud-native
Data Science/ML bridging
Research-to-production

Education

Master's degree or equivalent

Tools

AgentCore
LangGraph
AWS (S3, Lambda, ECS/EKS, Step Functions)
SageMaker/Other cloud ML platforms
Terraform
FastAPI

Job description

Minimum Qualifications
  • Master's degree in Computer Science, Software Development, Machine Learning, or a related field, OR equivalent professional experience (3-5+ years in production-level engineering).
  • Expert-level proficiency in Python with a focus on building distributed, scalable cloud-native services.
  • Proven experience in a Data Science or Machine Learning environment, specifically in bridging the gap between research code and production software.
Preferred Qualifications
  • Agentic AI & Orchestration: Hands‑on experience with AgentCore runtime for building and managing autonomous agents.
  • Extensive experience using LangGraph to create complex, stateful multi-agent orchestrations with high visibility.
  • Deep familiarity with Amazon Bedrock, OpenAI, or Anthropic APIs and the latest advancements in LLM reasoning.
  • Experience building and optimizing RAG (Retrieval-Augmented Generation) pipelines.
  • Machine Learning Engineering (MLE): Proven track record of productionizing Data Science models, transforming research-grade code into high-performance, scalable APIs (e.g., using FastAPI).
  • Experience with the full MLOps lifecycle: model deployment, versioning, and performance monitoring.
  • Familiarity with Amazon SageMaker or other cloud-based ML platforms.
  • Cloud & Infrastructure (AWS): Expertise in the AWS ecosystem: Lambda (Serverless), Step Functions, ECS/EKS (Containers), EventBridge, and S3.
  • Strong proficiency in Infrastructure as Code (IaC) using Terraform.
  • Experience building asynchronous, event-driven architectures.
  • Observability & Engineering Excellence: Proficiency in Splunk and CloudWatch for production monitoring and alerting.
  • Strong knowledge of software development life cycle (SDLC) processes, including unit testing, regression testing, and Agile concepts.
  • Ability to work with broad, loosely developed concepts and translate them into precise technical specifications.
Key Responsibilities
  • Agentic AI Innovation: Design and implement autonomous agents using AgentCore and orchestrate them via LangGraph to ensure complex workflows are visible and manageable.
  • MLE Productionization: Partner with Data Scientists to take ML models from research notebooks into scalable, production-ready AWS environments.
  • Constant Research: Proactively research, test, and present the newest technologies, frameworks, and AI research papers to the team. You are expected to be an early adopter of tools that can improve our velocity or service quality.
  • System Architecture: Build and maintain the cloud-native infrastructure (AWS) required for AI/ML inference and agentic execution, ensuring high availability and cost-efficiency.
  • Observability: Implement deep monitoring and alerting for all services, using LangGraph for agent-specific visibility and Splunk for broader system health.
  • Code Quality: Participate in rigorous code reviews and help define the engineering standards for the Data Science team.
  • Collaboration: Work without complete specifications to help derive technology solutions that meet the evolving needs of the business.
What you’ll get

Our team members fuel our strategy, innovation and growth, so we ensure the health and well-being of not just you, but your family, too! We go above and beyond to give you the support you need on an individual level and offer all sorts of ways to help you live your best life. We are proud to offer eligible team members perks and health benefits that will help you have peace of mind. Simply put: We’ve got your back.

Check out our full list of Benefits and Perks.

On-Call Expectations

This role may include participation in an on-call rotation to support production systems and ensure service reliability. On-call responsibilities may include coverage during nights and weekends. If applicable, frequency and scheduling will be determined by team needs and communicated accordingly.

About us

Rocket Close is a leading national provider of title insurance, property valuations and settlement services. Here, you’ll be given all the resources and support needed to deliver innovative solutions and in turn, your hard work will be rewarded with a competitive compensation package and an array of other amazing benefits.

This job description is an outline of the primary responsibilities of this position and may be modified at the discretion of the company at any time.

Decisions related to employment are not based on race, color, religion, national origin, sex, physical or mental disability, sexual orientation, gender identity or expression, age, military or veteran status or any other characteristic protected by state or federal law.

The company provides reasonable accommodations to qualified individuals with disabilities in accordance with applicable state and federal laws.

Applicants requiring reasonable accommodations in completing the application and/or participating in the application process should contact a member of the Human Resources team, at Careers@Rocket.com.

As part of the ecosystem of Rocket® we transform talented candidates into passionate team members who thrive in our world of opportunities.

Join us in our mission to shape the future of our industries and make a lasting impact in our communities and beyond.

Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

Software Engineer II - Python AI Engineer
Software Engineer II - Python AI Engineer

Rocket Companies • Detroit (MI)

On-site
USD 120,000 - 170,000
Health benefits
Software Engineer II - Python AI Engineer
Software Engineer II - Python AI Engineer

Rocket • Detroit (MI)

On-site
USD 140,000 - 190,000
Software Engineer II - Python AI Engineer
Software Engineer II - Python AI Engineer

Redfin Corporation • Detroit (MI)

On-site
USD 140,000 - 190,000
Senior Software Engineer I
Senior Software Engineer I

Rock Connections • Detroit (MI)

On-site
USD 120,000 - 160,000
Team Leader, Engineering
Team Leader, Engineering

Rocket Homes Real Estate LLC • Detroit (MI)

On-site
USD 140,000 - 190,000
Health benefits
Performance incentives
Career growth
Staff Software Engineer (Core Servicing Platform)
Staff Software Engineer (Core Servicing Platform)

Rocket Homes Real Estate LLC • Detroit (MI)

On-site
USD 150,000 - 210,000
Software Engineer I
Software Engineer I

Rocket • Detroit (MI)

On-site
USD 85,000 - 110,000
Lead Product Owner
Lead Product Owner

Rocket Homes Real Estate LLC • Detroit (MI)

On-site
USD 120,000 - 180,000
AI Engineer (Agentic Operations) - NLP
AI Engineer (Agentic Operations) - NLP

AIToolboard • United States

On-site
USD 120,000 - 180,000
Senior Systems Engineer (Hybrid or Remote)
Senior Systems Engineer (Hybrid or Remote)

Rocket Companies, Inc. • United States

Remote
USD 84,000 - 207,000