AI Engineer, Team Lead

Job Description

Join our team as an AI Engineer and Team Lead, where you'll be at the forefront of architecting and developing production-ready code for essential AI engagement features across Gravyty's innovative platforms. As a technical leader, you'll oversee the AI team, ensuring the delivery of high-quality projects with efficiency and operational excellence. Lead code reviews and mentor junior engineers, fostering a culture of technical proficiency within the Machine Learning team. Coordinate AI projects across product lines, aligning with institutional needs for recruitment, retention, and career readiness. You will build and optimize Retrieval-augmented generation (RAG) systems and agentic workflows to deliver personalized user support, and also oversee the deployment of MLOps systems to ensure the reliability and scalability of virtual assistants and predictive analytics tools. Additionally, you'll ensure all AI features adhere to enterprise-grade encryption and standards like SOC2 Type 2, FERPA, GDPR, and HIPAA.

Qualifications

1. 5+ years of experience in AI/ML engineering 2. 2+ years of experience managing a high-output ML team 3. Bachelor’s degree or higher in Machine Learning, Artificial Intelligence, Software Engineering, or a related field 4. Specialized expertise in designing and deploying Agentic AI systems and autonomous workflows 5. Mastery of Python, SQL, and TypeScript (Node.js) 6. Extensive experience with PyTorch, TensorFlow, HuggingFace, Scikit-learn, and Pandas 7. Proven expertise in LLMs, RAG, similarity search algorithms, and VectorDBs 8. Hands-on experience with LangChain or LangGraph 9. Proficiency with OpenAPI, FastAPI, and building scalable microservices 10. Deep knowledge of Docker, Git, CI/CD, and MLFlow 11. Significant experience deploying AI solutions on AWS or GCP