AI Engineer – Full Stack AI (Deep Learning, GNNs, NLP & Backend Integration)

A reputable company in Tehran province invites qualified people to apply.

Province
Tehran
Gender
Male, Female
Work Experience
Experience Required
Job Type
Remote

Requirements

  • We’re hiring an experienced, full-stack AI Engineer to independently build, integrate, and own the core of an AI-powered system focused on understanding and simulating computer network behavior — not just another chatbot or conversational AI project.
  • This is a contract role with strong potential for long-term collaboration and performance-based bonuses.

About the Role

  • You will be the sole AI/ML engineer and backend developer, responsible for the full development cycle of a platform that uses structured data (such as device configurations and topology models) to reason about computer networks using graph-based deep learning and natural language interpretation.
  • This is a highly technical, hands-on role. You must be comfortable working independently, solving complex infrastructure modeling problems, and producing clear, well-documented solutions.

Responsibilities

  • Design, build, and train Graph Neural Networks (GNNs) to analyze structured, graph-like data related to network topologies and configurations
  • Integrate and fine-tune Large Language Models (LLMs) (e.g., GPT, Claude, LLaMA) to interpret questions about network behavior
  • Parse structured files (YAML, JSON) and transform them into feature-rich graphs
  • Develop backend services and APIs to serve model predictions
  • Own the entire ML lifecycle — from data ingestion to deployment
  • Document all assumptions, architecture, models, and decisions
  • Be fully available during EST timezone working hours for daily standups, ad-hoc syncs, and collaborative discussions

Required Skills

  • 3+ years of hands-on AI/ML engineering experience
  • Strong Python programming skills and experience with PyTorch (or TensorFlow)
  • Experience building Graph Neural Networks using frameworks like PyTorch Geometric or DGL
  • Familiarity with LLM integration (OpenAI API, HuggingFace Transformers, LangChain, etc.)
  • Backend experience (FastAPI, Flask, or similar) for building APIs to expose model outputs
  • Excellent problem-solving skills and the ability to work autonomously
  • Clear, consistent documentation habits
  • Full-time availability Sunday through Friday during EST hours

Nice to Have

  • Background or strong interest in computer networking (e.g., routing protocols, topologies, infrastructure configuration)
  • Experience working with structured network data, device models, or configuration formats
  • Familiarity with Docker, CI/CD, or ML Ops tools
  • Interest in long-term technical ownership or evolving into a lead role

Engagement Details

  • Fully remote
  • Must be available Sunday–Friday, during EST business hours
  • Performance-based bonuses available
  • Potential for long-term collaboration based on success and alignment
  • This role is focused on computer networking behavior modeling, not conversational chatbots or standard NLP products