Job Title: AI/ML Engineer – Python & Agentic AI Systems
Location: Noida / Chandigarh / Gurugram, India (Hybrid – 2 days WFO)
Employment Type: Full-time
Experience: 6+ Years
About the Role
We are looking for a highly skilled AI/ML Engineer to design, build, and scale intelligent AI-driven systems using Python, RESTful APIs, and AWS AI/ML services.
This role demands deep expertise in LLMs, agentic workflows, and document extraction systems, along with strong architectural decision-making abilities across performance, cost, accuracy, and scalability.
The ideal candidate is hands-on, technically strong, and capable of driving design discussions, reviewing complex codebases, and mentoring engineers while delivering production-grade AI solutions.
Key Responsibilities
- Design and develop scalable, high-performance Python services
- Build and maintain REST APIs using Python
- Implement asynchronous processing for I/O-heavy and compute-intensive workloads
- Design and implement LLM-based and agentic workflows using AWS AI services (e.g., Amazon Bedrock, SageMaker)
- Architect document and data extraction pipelines for structured and unstructured data (PDFs, etc.)
- Evaluate and communicate technical trade-offs and architectural decisions
- Debug, optimize, and productionize AI/ML systems
- Review and refactor complex Python codebases ensuring maintainability and type safety
- Collaborate with cross-functional teams to convert business requirements into scalable solutions
Required Skills & Experience
- Strong hands-on experience in Python development with production-grade coding practices
- Experience in building and maintaining REST APIs
- Expertise in async programming (asyncio / async I/O patterns)
- Strong focus on type safety (typing, Pydantic or equivalent)
- Ability to understand and refactor complex Python codebases
- Hands-on experience with AWS AI/ML services (Amazon Bedrock, SageMaker, AI Builder tools)
- Experience with Amazon Textract or similar document AI tools
- Deep understanding of LLMs (context limits, latency, cost, hallucination, evaluation)
- Experience building agentic workflows (tool-calling, orchestration, multi-step flows)
- Strong experience in document extraction systems (PDF/unstructured → structured data)
- Knowledge of PDF extraction challenges (OCR, scanned vs digital, tables, layouts)
- Ability to analyze and justify technical trade-offs (accuracy vs latency, cost vs capability, etc.)
- Strong debugging and performance optimization skills
- Familiarity with logging, monitoring, and production troubleshooting on AWS
Soft Skills
- Strong analytical and problem-solving skills
- Clear communication of technical concepts and trade-offs
- Ownership mindset with focus on quality and reliability
- Ability to work with both technical and non-technical stakeholders
- Mentorship and leadership capabilities
- Comfortable working in fast-evolving AI environments
Nice-to-Have
- Experience with multi-agent architectures
- Exposure to LLM evaluation frameworks
- Knowledge of AI cost optimization strategies
- Familiarity with cloud-native AWS architectures
- Experience with enterprise-scale document workflows
Education
- Bachelor's or Master's degree in Computer Science, Engineering, or a related field
- Equivalent practical experience will also be considered
