AI Engineer (Python, Machine Learning & Generative AI)
ATCAbout the role
Job Title: AI Engineer (Python, Machine Learning & Generative AI)
Experience developing REST APIs using FastAPI or Flask.
We are looking for an AI Engineer with 3+ years of experience in Artificial Intelligence, Machine Learning, and Generative AI. The ideal candidate will have hands-on experience developing AI-powered applications using Python, Large Language Models (LLMs), LangChain, LangGraph, and Retrieval-Augmented Generation (RAG). You will work on designing, building, and deploying scalable AI solutions for enterprise applications.
Key Responsibilities
● Develop and deploy AI and Machine Learning solutions using Python.
● Build LLM-powered applications using LangChain and LangGraph.
● Design and implement Retrieval-Augmented Generation (RAG) pipelines with vector databases.
● Develop AI agents and conversational AI applications.
● Train, evaluate, and optimize machine learning and deep learning models.
● Build REST APIs using FastAPI or Flask to integrate AI services.
● Integrate AI applications with enterprise systems, databases, and third-party APIs.
● Work with structured and unstructured data for model training and inference.
● Deploy AI applications using Docker and cloud platforms such as AWS, Azure, or GCP.
● Collaborate with cross-functional teams to deliver AI-driven business solutions.
● Troubleshoot, optimize, and maintain AI applications in production.
Required Qualifications
● Master’s degree in Computer Science, Artificial Intelligence, Data Science, or a related field.
● 3+ years of experience in AI, Machine Learning, or Software Development.
● Strong programming skills in Python.
● Hands-on experience with TensorFlow, PyTorch, or Scikit-learn.
● Experience building applications with Large Language Models (LLMs).
● Experience with LangChain and LangGraph.
● Hands-on experience implementing Retrieval-Augmented Generation (RAG).
● Experience with vector databases such as Pinecone, ChromaDB, FAISS, or Milvus.
● Knowledge of SQL and NoSQL databases.
● Experience with Git, Docker, and CI/CD pipelines.
● Familiarity with cloud platforms (AWS, Azure, or Google Cloud Platform).
Preferred Qualifications
● Experience with OpenAI, Anthropic, or open-source LLMs.
● Knowledge of prompt engineering, function calling, and AI agent frameworks.
● Familiarity with Hugging Face Transformers and LlamaIndex.
● Experience with Kubernetes and MLOps tools such as MLflow or Kubeflow.
● Exposure to NLP, deep learning, or computer vision.
Responsibilities
- Develop and deploy AI and Machine Learning solutions using Python
- Build LLM-powered applications using LangChain and LangGraph
- Design and implement Retrieval-Augmented Generation (RAG) pipelines with vector databases
- Develop AI agents and conversational AI applications
- Train, evaluate, and optimize machine learning and deep learning models
- Build REST APIs using FastAPI or Flask to integrate AI services
- Integrate AI applications with enterprise systems, databases, and third-party APIs
- Work with structured and unstructured data for model training and inference
Qualifications
- Master’s degree in Computer Science, Artificial Intelligence, Data Science, or a related field
- 3+ years of experience in AI, Machine Learning, or Software Development
- Strong programming skills in Python
- Hands-on experience with TensorFlow, PyTorch, or Scikit-learn
- Experience building applications with Large Language Models (LLMs)
- Experience with LangChain and LangGraph
- Hands-on experience implementing Retrieval-Augmented Generation (RAG)
- Experience with vector databases such as Pinecone, ChromaDB, FAISS, or Milvus
Skills mentioned
About ATC
Founded in the wake of the 2008 Great Recession, ATC has evolved into a dynamic tech company, known for its ability to adapt quickly to the ever-changing business landscape. For over a decade, we’ve been at the forefront of technological innovation, helping companies transform challenges into opportunities. In a world where innovation is essential, ATC empowers businesses to stay ahead by creating cutting-edge software solutions that drive productivity and outperform the competition. We go beyond conventional software development to solve complex business problems with creativity and a passion for technology that enriches lives. Our comprehensive offerings span strategy, design, automation, custom software development, managed services, and advanced IT solutions, including chatbots and cloud transformations. Through ATC Ventures, we support visionary entrepreneurs, helping them build groundbreaking SaaS startups from the ground up. Our commitment to technology extends to our Training Division, which has trained over 10,000 professionals across various IT disciplines. As a proud Scaled Agile Gold Transformation Partner, we provide expert training, coaching, and consulting services. With headquarters in Iowa and delivery centers across the U.S. and India, our talented, dedicated team is driven to bring bold ideas to life. ATC specializes in a wide range of services, including: Enterprise AI Cybersecurity DevSecOps Staffing Software Testing & Automation SaaS Product Development Product Design (UX/UI) Custom AI Models Public Sector Services (State and Federal) At ATC, we don’t just build software – we create solutions that shape the future.