Back to Careers
3 to 5 Years Experience

Python Developer (AI/LLM)

Full-time
3 to 5 Years Experience
Remote

Overview

We're seeking a Python Developer to join our AI engineering team. You'll build end-to-end RAG pipelines, fine-tune LLMs with vLLM, architect Neo4j graph databases, and develop Python microservices — powering the intelligent features of our virtual data room platform.

Key Responsibilities

AI & LLM Systems

  • Build end-to-end RAG (Retrieval-Augmented Generation) pipelines for context-aware AI responses.
  • Implement and fine-tune vLLM for efficient inference of large language models (LLMs).
  • Collaborate with ML engineers to deploy transformer models (e.g., BERT, GPT variants) and vector databases.

Data & Database Architecture

  • Architect and optimize graph database systems (Neo4j) to model project knowledge networks and relationships.
  • Develop Python-based microservices for data ingestion, processing, and API integrations (FastAPI, Flask).

Performance & Operations

  • Monitor system performance, conduct A/B tests, and ensure low-latency responses in production.
  • Ensure scalability and efficiency of AI systems.

Requirements

  • Proficiency in Python and AI/ML libraries (PyTorch, TensorFlow, Hugging Face Transformers).
  • Hands-on experience with graph databases, especially Neo4j (Cypher queries, graph algorithms).
  • Demonstrated work on RAG pipelines (retrieval, reranking, generation) using frameworks like LangChain or LlamaIndex.
  • Experience with vLLM or similar LLM optimization tools (quantization, distributed inference).
  • Knowledge of vector databases (e.g., FAISS, Pinecone) and embedding techniques.
  • Familiarity with cloud platforms (AWS/GCP/Azure) and containerization (Docker, Kubernetes).

Preferred Qualifications

  • Strong experience with FastAPI or Flask for building high-performance APIs.
  • Familiarity with MLOps principles and tools (e.g., MLflow, Kubeflow).
  • Contributions to open-source AI/ML projects.
  • Experience in performance tuning and A/B testing for AI systems in a production environment.

Soft Skills

  • Strong analytical and problem-solving skills.
  • Excellent communication and team collaboration abilities.
  • Self-motivated with the ability to work independently and as part of a team.

What We Offer

  • Competitive salary and performance-based bonuses.
  • Flexible working hours with remote work options.
  • Opportunities for professional development and skill enhancement.
  • Collaborative and inclusive work environment.
  • Paid sick time
  • Paid time off
  • Provident Fund
  • Performance bonus
  • Yearly bonus

Don't see a position that fits your skills?

Send us your resume