Position : Python AI Developer | Experience : 5+ years | Location : Ahmedabad | Work mode : On-site | Working days: 5 days

This position offers a significant opportunity to impact a product that is evolving within the legal industry and is a leading use of AI in this sector (a sector that is very well suited to the strengths of Gen AI). The successful candidate will join a dynamic team, addressing complex technical challenges and contributing directly to the expansion of our AI solutions.

What the Role Involves:

● Working on the scalability, integration, and deployment of our core AI model and associated backend services.

● Developing, and maintaining high-performance, reliable Python-based applications utilising FastAPI and Pydantic.

● Extensive work with Postgres RDS and contributing to the continuous development of our knowledge graph.

● Implementing and optimising solutions for containerised environments, leveraging AWS ECS and Fargate.

● Collaborating with AI/ML specialists to integrate advanced AI components (e.g., Graph-based Retrieval-Augmented Generation (RAG), LangChain, Llama Index) and external model inference APIs into the core platform.

● Defining build and test scripts for developed code, working in conjunction with our dedicated DevOps team who manage full CI/CD pipelines.

● Instantiating necessary simple AWS components via the AWS admin UI or CLI, ensuring smooth operation and integration.

● Work effectively in a fast-paced, collaborative setting with engineering colleagues and also with product managers

● Proactively working in the resolution of outages or technical issues where required candidate

Requirements

● 5+ years software development experience in Python.

● Strong proficiency in Python and extensive experience with modern frameworks such as FastAPI and Pydantic.

● Good experience with data engineering principles and working with relational databases, particularly Postgres,



SQL and ORMs. Knowledge of caches like Redis is an advantage but not required.

● Knowledge of Python internals and technicalities, performance and concurrency

● Experience with AWS services, specifically EC2/ECS, Lambda Functions, S3 and RDS.

● Good knowledge of software engineering best practices, including robust testing, code reviews, git branch strategies and management and version control.

● Experience with containerisation (e.g., Docker) and CI. Having worked with Github Actions and Kubernetes/EKS is an advantage but not required.

● Knowledge of Gen AI concepts and their Python libraries such as Retrieval-Augmented Generation (RAG), LangChain, or Llama Index is desirable but not required (although a motivation to get on the leading edge of it is needed).

● Experience with Python machine learning libraries like NumPy, Pandas, Scikit Learn, Huggingface and the Onnx Runtime are an advantage but not required

● May have some experience with full stack development or web development, even if not the preferred area to work in. React/NextJS experience a bonus - not expected to be a significant part of the work of the role, but being able to understand the code and work with it when needed independently will make the candidate more efficient

● Excellent problem-solving skills and a proactive approach to technical challenges.

● Strong communication skills and capability to provide transparent proactive communications of problems, solutions, options and investigations

Our Technology Stack:

● Backend & AI Integration: Python (FastAPI, Pydantic), Postgres RDS, Knowledge Graphs

● Cloud Platform: AWS (EC2/ECS, Lambda Functions, S3)

● AI/ML: Graph-based Retrieval-Augmented Generation (Graph RAG), LangChain, OnnxRuntime. Many different LLMs being used for specific purposes as part of a composite system.

● Infrastructure: Containerised architecture on Kubernetes/EKS

● Admin tool: NextJS and ShadCnUI bespoke admin tool

📌 Python AI Developer (Ahmedabad)
🏢 JDS Advisory
📍 Ahmedabad