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.