Rocket Companies

Software Engineer II - Python AI Engineer

Rocket Companies
Apply
2 days ago
Detroit, MI, USAMid Level / Senior
H1B sponsor

Responsibilities

  • Design and implement autonomous agents with AgentCore and orchestrate multi-agent workflows using LangGraph.
  • Transform Data Science models and research code into scalable, production-ready AWS services and APIs.
  • Research, test, and present emerging AI technologies, frameworks, and research papers.
  • Build and maintain highly available, cost-efficient AWS infrastructure for AI/ML inference and agent execution.
  • Implement service monitoring, alerting, and agent-specific visibility using Splunk, CloudWatch, and LangGraph.
  • Participate in code reviews, define engineering standards, and derive technical solutions from incomplete specifications.
  • Participate in an on-call rotation when applicable, including potential nights and weekends.

Requirements

  • Master’s degree in Computer Science, Software Development, Machine Learning, or a related field, or equivalent professional experience.
  • At least 3 years of production-level engineering experience, with the posting describing a 3–5+ year range.
  • Expert Python proficiency focused on distributed, scalable, cloud-native services.
  • Experience in Data Science or Machine Learning environments and bridging research code to production software.
  • Hands-on experience with AgentCore and extensive experience with LangGraph for agentic orchestration.
  • Familiarity with Amazon Bedrock, OpenAI, or Anthropic APIs, LLM reasoning, and RAG pipelines.
  • Experience productionizing machine learning models into scalable APIs, such as with FastAPI.
  • Experience with model deployment, versioning, performance monitoring, and the MLOps lifecycle.
  • Familiarity with Amazon SageMaker or other cloud-based ML platforms.
  • Expertise with AWS services including Lambda, Step Functions, ECS/EKS, EventBridge, and S3.
  • Proficiency with Terraform and asynchronous, event-driven architectures.
  • Proficiency with Splunk and CloudWatch for production monitoring and alerting.
  • Knowledge of unit testing, regression testing, software development lifecycle processes, and Agile concepts.
  • Ability to translate broad, loosely developed concepts into precise technical specifications.

Benefits

  • Eligible team members receive health and family well-being benefits and other perks.
  • The company offers career growth opportunities and a competitive compensation package.
  • The role may involve on-call coverage during nights and weekends, with frequency determined by team needs.
Rocket Companies

About Rocket Companies

10,000+ employees

Own the Dream. Mortgage services provided by Rocket Mortgage, LLC. NMLSConsumerAccess.org, NMLS #3030

Contact me