Trainee/Junior AI Specialist (Python)

AI Specialist (Python)

We are looking for an AI Specialist with strong Python skills and hands-on experience working with modern AI frameworks and LLMs. You will work on building, fine-tuning, and deploying machine-learning models into production, as well as developing Python backend services for model serving.

Responsibilities

• Develop and maintain AI/ML solutions using Python.
• Work with LLMs (OpenAI, Anthropic, Google Gemini) and frameworks like LangGraph and LangChain.
• Build and optimize pipelines using Vector Stores.
• Train, fine-tune, and evaluate ML models (image, text, audio, video).
• Work with neural networks, NLP, computer vision or reinforcement learning tasks.
• Create and improve classification models (shot/scene detection, captions, object labeling, attention scoring, etc.).
• Deploy AI/ML models into production environments.
• Develop Python backend (Flask or FastAPI) for serving models.
• Work with databases (SQL and NoSQL) for data storage, preprocessing, and model input pipelines.

Requirements

• >1 year of experience in AI/ML or applied machine learning.
• Strong Python skills.
• Experience with LLM frameworks: LangGraph, LangChain.
• Hands-on experience with LLMs (OpenAI, Anthropic, Gemini).
• Good knowledge of Vector Stores and their application.
• Solid understanding of ML algorithms, architectures, and workflows.
• Experience with ML frameworks: PyTorch, TensorFlow.
• Experience with image/text/audio/video model training and fine-tuning.
• Experience building backend services with Flask or FastAPI.
• Experience working with SQL and NoSQL databases (e.g., PostgreSQL, MySQL, MongoDB, DynamoDB, etc.).

Nice to have

• Understanding of MLOps tools and CI/CD for ML.
• Experience with GPUs, model optimization, quantization.
• Familiarity with cloud platforms (AWS / GCP / Azure).