
Software Engineer II - Python AI Engineer
Rocket Companies2 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.
Categories
About Rocket Companies
Own the Dream. Mortgage services provided by Rocket Mortgage, LLC. NMLSConsumerAccess.org, NMLS #3030