Photon
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.
- 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, aio
- 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
- 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.
- 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.
- Bachelor's/Master's degree in Computer Science, Engineering, or related field.
- Strong track record of building scalable frameworks or platforms in production environments.