ML/AIWork
Photon logo

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 →
Python Developer with ML - Dallas, TX
Photon
Apply →