Python Developer (AI / ML)

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

We are hiring 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.

SpaceLizit is building the AI layer for US immigration law — automating petitions, predicting RFE risk, and enabling attorneys to process 10x more cases with AI-driven workflows.

Key Responsibilities

1 AI / ML Systems

• 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

2 Graph Database & Knowledge Systems

• Model complex immigration case relationships in Neo4j knowledge graphs
• Write optimised Cypher queries for multi-hop relationship traversal
• Build knowledge extraction pipelines from USCIS forms and legal documents
• Design graph-based reasoning systems for case eligibility assessment

3 Backend & API Development

• Build production-grade Python APIs using FastAPI or Django REST Framework
• Integrate AI services with the core MERN stack via gRPC or REST
• Implement async task queues using Celery and Redis
• Containerise services with Docker and deploy on Kubernetes / AWS ECS

4 Data Engineering

• Build ETL pipelines for processing USCIS forms, PDFs, and legal documents
• Implement OCR and document intelligence using Amazon Textract or Azure Form Recognizer
• Monitor model performance metrics and implement continuous retraining workflows
• Maintain data quality, lineage, and governance standards

Requirements

• 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

Preferred Qualifications

• Experience with vLLM, TensorRT, or ONNX for model optimisation
• Knowledge of Apache Kafka or similar event streaming platforms
• Background in NLP, information extraction, or document intelligence
• Familiarity with USCIS forms, immigration processes, or legal-tech
• Open-source AI project contributions

Soft Skills

• Strong analytical and systems-thinking mindset
• Ability to translate complex AI concepts for non-technical stakeholders
• Self-directed with the ability to navigate ambiguity in research-driven work
• Collaborative approach to cross-functional engineering
• Clear written and verbal communication in English

What We Offer

• Fully remote — global team across US, India, Singapore, and UAE
• Competitive salary benchmarked against global AI engineering rates
• Direct access to frontier AI research and tooling
• Annual learning & conference budget
• Fast-track impact: your AI directly affects thousands of immigration cases
• Flexible working hours with async-first culture
• Performance bonuses tied to product impact metrics

Application Questions

1. Describe an RAG system you've built — what retrieval strategy, embedding model, and LLM did you use?

2. Have you worked with graph databases? Describe a graph data model you designed and why.

3. What is your experience with LLM fine-tuning vs prompt engineering — when do you choose each?

Ready to Apply?

Send your resume and a brief cover letter explaining why you're a great fit. We review every application personally within 48 hours.