Python AI/ML Engineer - RAG & Graph Analytics (Remote)

SpaceLizit; United States; Remote; USD 120,000 - 180,000; Full time

Benefits offered by this job

Fully remote

Learning budget for courses and events

Performance bonuses

Flexible working hours

Frontier AI research access

Job summary

SpaceLizit is seeking a Python Developer with strong AI/ML expertise to design and scale our intelligent document processing and legal reasoning systems. You will work on RAG pipelines, graph databases, and LLM optimization to power SpaceLizit's core AI capabilities that automate complex immigration workflows for US law firms.

Join a fully remote, globally distributed team building AI that automates immigration petitions, predicts RFE risk, and enables attorneys to process more cases with

Qualifications

• 3–5 years of professional Python development experience.
• Hands-on experience building production RAG or LLM-based systems.
• Strong proficiency with LangChain, LlamaIndex, or similar AI orchestration frameworks.
• Experience with Neo4j or other graph databases.
• Solid understanding of vector embeddings and semantic search.
• Proficiency with FastAPI or Django REST Framework.
• Experience with Docker, Kubernetes, and cloud platforms (AWS/GCP/Azure).
• Strong algorithmic and data structures fundamentals.

Responsibilities

• Design and implement Retrieval-Augmented Generation (RAG) pipelines for legal document processing.
• Optimise LLM inference using vLLM, TensorRT, or similar frameworks.
• Build and maintain vector search systems (Pinecone, Weaviate, or pgvector).
• Develop prompt engineering strategies for immigration-specific language models.

Skills

Python development

AI/ML systems

LangChain

LlamaIndex

Neo4j

vector embeddings

FastAPI

Django REST Framework

Docker

Kubernetes

AWS

GCP

Azure

Tools

Neo4j

FastAPI

Django REST Framework

Docker

Kubernetes

AWS

GCP

Azure

Celery

Redis

gRPC

Job description

SpaceLizit is seeking a Python Developer with strong AI/ML expertise to design and scale our intelligent document processing and legal reasoning systems. You will work on RAG pipelines, graph databases, and LLM optimization to power SpaceLizit's core AI capabilities that automate complex immigration workflows for US law firms.

Join a fully remote, globally distributed team building AI that automates immigration petitions, predicts RFE risk, and enables attorneys to process more cases with