Python Developer with ML - Dallas, TX
Photon · Dallas, US
Job description
We are building next-generation intelligent systems powered by AI and automation. Our team is looking for a skilled Python Developer for ML with deep expertise in framework design and exposure to Agentic AI systems. You will play a pivotal role in architecting scalable frameworks, APIs, and integrations while enabling AI-driven agents to perform complex workflows securely and efficiently.
Key Responsibilities:
-
Design, develop, and maintain Python frameworks that provide reusable components and structure for applications.
-
Build, train, and deploy robust machine learning and deep learning models using Python and popular frameworks (e.g., PyTorch, TensorFlow).
-
Architect modular, scalable, and extensible frameworks for APIs, data processing, or AI integrations.
-
Implement best practices in software engineering: OOP, design patterns, modular architecture.
-
Develop and optimize Retrieval-Augmented Generation (RAG) pipelines for Large Language Models (LLMs) to provide contextually relevant and accurate outputs. This includes ingesting, processing, and indexing data from various sources (e.g., PDFs, HTML, databases).
-
Develop APIs and SDKs for framework adoption across teams.
-
Utilize vector databases (e.g., Pinecone, Weaviate) for efficient storage and retrieval of high-dimensional vector embeddings to support RAG and semantic search functionalities.
-
Collaborate with AI engineers to integrate Agentic AI systems (e.g., AI agents, LLM orchestration frameworks like LangChain, LlamaIndex, Google ADK).
-
Ensure security, reliability, and observability of framework components.
-
Write unit/integration tests and maintain CI/CD pipelines.
-
Document frameworks, APIs, and libraries for developer adoption.
-
Participate in code reviews, mentor junior developers, and contribute to technical design discussions. Required Skills & Experience:
-
Strong proficiency in Python (3.x) for ML and software development., with focus on OOP and modular design.
-
Proven experience in building frameworks, libraries, or SDKs (not just applications).
-
Advanced proficiency in Python for ML and software development.
-
Expertise in design patterns, dependency injection, and plugin-based architectures.
-
Solid understanding of ML algorithms, model evaluation techniques, and MLOps practices.
-
Experience developing with Large Language Models (LLMs) and advanced Prompt Engineering techniques.
-
Experience with FastAPI, Flask, Django or similar web frameworks.
-
Solid understanding of APIs (REST, GraphQL) and API gateway integration.
-
Knowledge of async programming (asyncio, aiohttp).
-
Hands-on experience building and deploying Retrieval-Augmented Generation (RAG) systems.
-
Proficiency in testing frameworks (PyTest, unittest) and CI/CD workflows.
-
Strong grasp of security practices (OAuth2, JWT, IAM integration).
-
Familiarity with cloud platforms (GCP, Azure, AWS) for deploying frameworks.
-
Familiarity with vector databases for efficient similarity search Nice to Have:
-
Experience with Agentic AI frameworks (OpenAI SDK, LangChain, LlamaIndex, Google ADK, AutoGen).
-
Knowledge of LLM fine-tuning, prompt engineering, and guardrails.
-
Exposure to event-driven architectures (Kafka, Pub/Sub, Redis Streams).
-
Experience with observability tools (Prometheus, OpenTelemetry, Cloud Logging).
-
Contributions to open-source Python frameworks or libraries. Ideal Candidate:
-
A framework builder mindset – you enjoy designing systems other developers will use.
-
Passionate about AI and automation, eager to experiment with agent-driven architectures.
-
A problem-solver with a strong understanding of scalable and secure design principles.
-
Excellent communicator with strong documentation and collaboration skills. Qualifications
-
Bachelor’s/Master’s degree in Computer Science, Engineering, or related field.
-
Strong track record of building scalable frameworks or platforms in production environments.
ML/AI Work links you to the employer's original posting — always verify the details there before applying.
More MLOps and Platform roles
View all →Machine Learning Engineer, Generative ML , Level 5
Snap Inc. · Anaheim, US
Director, AI Engineering
Menarini · Remote · New York
Senior Platform Engineer - AI
Datavant · Remote · New York
Director - AI Platform Engineering
eBay · Remote · San Jose
Senior Data Scientist (TS/SCI with CI Poly Required)
DeNovo Solutions · Aurora, US
Project Lead AI
Entico · Charleroi, BE