Staff AI/ML Engineer: LLMs, LangChain, Python (Remote)

United States Digital Space LLC; United States; Remote; USD 120,000 - 150,000; Full time

Benefits offered by this job

Equity compensation

Performance-based bonuses

Remote-first workforce

Health insurance reimbursements up to

401(k) with company match

Job summary

United States Digital Space LLC is seeking a Member of Technical Staff, AI/ML Engineering, to drive cutting-edge AI systems in a remote-first environment. You will design, implement, and optimize LLM-based solutions using Python, AWS, LangChain, and FastAPI, collaborating with engineers to translate complex mathematical and physics-based problems into scalable products.

The role emphasizes strong math/quantitative background, production-ready ML workflows, and clear communication of progress.

Qualifications

• Advanced degree or demonstrable expertise in math, quantitative sciences, statistics, or physics.
• Expert proficiency in Python for AI/ML engineering.
• Hands-on experience deploying, fine-tuning, and integrating LLMs.
• Strong background in AWS services and cloud-native ML workflows.
• Proven track record using LangChain for chaining LLMs and orchestrating multi-step reasoning.
• Solid understanding of FastAPI for building scalable APIs.
• Exceptional written and verbal communication skills; ability to articulate complex technical topics with clarity.

Responsibilities

• Design, build, and deploy AI/ML solutions with a focus on large language models (LLMs) using Python and AWS infrastructure.
• Architect scalable APIs and microservices utilizing FastAPI, ensuring robust, high-performance systems.
• Integrate LangChain and advanced AI frameworks to deliver innovative model interaction capabilities.
• Collaborate with cross-functional teams to translate complex mathematical, statistical, or physics-based problem statements into actionable engineering solutions.
• Apply core math, statistics, or physics knowledge to improve model performance, interpretability, and reliability.
• Maintain high standards of code quality, documentation, and peer reviews within a fast-paced core team.
• Communicate technical concepts and project progress clearly through both written reports and verbal discussions.

Skills

Python

LLMs

AWS

LangChain

FastAPI

Education

Advanced degree in math/quantitative sciences/physics

Job description

United States Digital Space LLC is seeking a Member of Technical Staff, AI/ML Engineering, to drive cutting-edge AI systems in a remote-first environment. You will design, implement, and optimize LLM-based solutions using Python, AWS, LangChain, and FastAPI, collaborating with engineers to translate complex mathematical and physics-based problems into scalable products.

The role emphasizes strong math/quantitative background, production-ready ML workflows, and clear communication of progress.