Data Scientist

Role Overview

Atain is looking for a highly capable Data Scientist who can build and ship advanced AI solutions with a focus on machine learning, generative AI, and large language models. This role involves creating scalable models and enterprise-ready GenAI applications that solve real business problems.

Key Responsibilities

• Develop, train, and deploy machine learning models to address business challenges.
• Create generative AI solutions powered by large language models such as GPT, LLaMA, and similar tools.
• Design and improve retrieval-augmented generation pipelines.
• Work on prompt design, model fine-tuning, and performance tuning.
• Build AI agents and automation flows using frameworks such as LangChain and LangGraph.
• Handle data cleaning, preprocessing, feature creation, and model assessment.
• Partner with cross-functional teams to convert business requirements into AI solutions.
• Release models to production through cloud services and APIs.
• Continuously track, test, and enhance model quality after deployment.

Technical Skills and Experience

• Strong foundation in machine learning and deep learning.
• Advanced Python skills, including NumPy, Pandas, Scikit-learn, and PyTorch or TensorFlow.
• Practical exposure to large language models and generative AI systems.
• Experience working with OpenAI, Hugging Face, or comparable ecosystems.
• Good understanding of RAG design, embeddings, and vector databases.
• Hands-on experience with prompt engineering and fine-tuning LLMs.
• Exposure to agentic AI frameworks such as LangChain and LangGraph.
• Working knowledge of SQL, ETL workflows, and data processing.
• Preferred familiarity with Databricks or Spark.
• Knowledge of vector databases such as FAISS, Pinecone, and ChromaDB.
• Experience deploying solutions on Azure, AWS, or GCP.
• Understanding of production deployment practices including APIs, Docker, and CI/CD.

Application Note

Only candidates available to join immediately should share their profiles at the provided email address.