Job Description

Job Description

Requirements

  • Experience : 6+ Years
  • Strong proficiency in Python (OOP, data structures, async programming).
  • Experience with Generative AI / Deep Learning:
  • Transformers (GPT, BERT, T5,
  • Diffusion models (Stable Diffusion, — optional for some roles
  • Tokenizers and embeddings
  • Frameworks/Libraries:
  • PyTorch and/or TensorFlow
  • Hugging Face Transformers, Diffusers
  • NumPy, Pandas, Scikit-learn
  • Model training & fine-tuning on large datasets.
  • Building inference APIs (FastAPI/Flask).
  • Knowledge of GPU training and acceleration (CUDA, cuDNN).
  • Version control (Git), Collaboration workflows (PRs, Code reviews).

    • Responsibilities

  • Design, develop, and optimize generative AI models (., GPT-style, BERT, T5, Stable Diffusion).
  • Implement and maintain data pipelines for training and fine-tuning models.
  • Write high-quality, scalable Python code leveraging frameworks like PyTorch, TensorFlow, JAX, and Hugging Face.
  • Collaborate with product, design, and engineering teams to understand requirements and deploy AI features into production.
  • Build REST APIs and services for model inference (FastAPI, Flask, Django).
  • Monitor and maintain deployed models (logging, versioning, performance tracking).
  • Research and experiment with new AI techniques and tools to improve model performance.
  • Document model design, APIs, and deployment procedures

    • Qualifications

    Bachelor’s or master’s degree in computer science, Information Technology, or a related field.

    Apply for this Position

    Ready to join ? Click the button below to submit your application.

    Submit Application