Tech Lead AI and ML Engineer
Summary
Lead a team to design and build AI-powered Python libraries on AWS, using SageMaker and Bedrock while managing infrastructure with Terraform.
Job Title: Tech Lead AI and ML Engineer
Key Skills: Machine Learning, LLMs, Python, AWS
Location: Brazil
Mode: Remote
We at Coforge are hiring Tech Lead AI and ML Engineer with the following skill set.
Key Responsibilities
- Raise and discuss trade-offs of different design approaches elaborating detailed design documents;
- Improve and Abstract current functionalities into reusable Libraries;
- Develop solution components in partnership with the Data team;
- Participate in technical discussions with customer operations and Data team;
- Follow Agile processes and participate actively in all development phases, fostering a culture of collaboration;
- Lead a team of highly skilled developers into building cutting edge AI solutions.
Required Skills & Qualifications
- Experience with Python libraries for machine learning (Pandas, Matplotlib, Scikit learn, PyTorch);
- Experience with Machine Learning and Deep Learning;
- Experience with NLP (NLTK, tokenization, feature extraction, embeddings);
- Experience with LLMs (prompt engineering, few shot learning, LiteLLM API + AWS Bedrock);
- Experience with AWS Cloud (S3, Lambda, IAM, EC2, RDS, Cloudwatch, Bedrock, EKS);
- Experience with Distributed Systems;
- Experience with API implementation in Python (FastAPI, Pydantic, HTTP exceptions management);
- Experience in Leading Software Development Teams;
- Git version control;
- Linux environment;
- Experience with Jupyter Notebooks;
- Experience working with machine learning models in production;
- Software development expertise in general (unit test implementation, code review, clean code implementation, version control, etc);
- Experience in using GenAI in the software development lifecycle;
- Experience with ML models management (ex. MLFlow).
Preferred Skills:
- API Gateway concepts;
- AWS Bedrock experience;
- Creating Python Libraries;
- Experience with Data Science;
- Experience with MLOps.
Posted On: 18-08-2026
At Coforge, we hire professionals based solely on their skills and do not discriminate based on age, disability, religion, gender, sexual orientation, socioeconomic status, or nationality.