AI Engineer - SDE 2

Attri; Ahmedabad District; On-site; INR 800,000 - 1,200,000; Full time

Benefits offered by this job

Health Insurance

Professional development courses sponsorship

Flexible working hours

Friendly work environment

No Micromanagement

Job summary

A leading AI solutions company in Ahmedabad is seeking a Software Development Engineer II to develop and deploy scalable AI models and integrate them into core business systems. The ideal candidate will have strong Python skills, experience with REST APIs, and familiarity with cloud deployments. This role offers a collaborative work environment and supports professional development.

Qualifications

• Strong proficiency in Python for backend development.
• Hands-on experience integrating and consuming model inference APIs.
• Good backend fundamentals including SQL/NoSQL.

Responsibilities

• Develop and deploy scalable AI models.
• Integrate AI models into business systems.
• Collaborate with a specialized AI/ML engineering team.

Skills

Python

REST APIs

Microservices

Websockets

Context Engineering

Cloud Deployment

SQL/NoSQL

Tools

FastAPI

Docker

Kubernetes

Job description

Attri AI is at the forefront of developing innovative, production‑ready AI solutions. We are looking for an ambitious and skilled Software Development Engineer II to join our specialized AI/ML engineering team. This role is crucial for developing and deploying scalable, production‑ready AI models and integrating them into core business systems. The ideal candidate will have a strong foundation in Python, modern MLOps principles, and experience transforming prototypes into robust, high‑performance applications.

Core Requirements – SDE-2 Backend + AI Engineer

• Strong proficiency in Python for backend development (FastAPI / Flask / Django).
• Solid understanding of websockets, REST APIs, microservices, and distributed backend fundamentals.
• Hands‑on experience integrating and consuming model inference APIs (OpenAI, Anthropic, Google, local models, etc.).
• Practical experience with agentic frameworks such as:
• LangGraph
• OpenAI Agents SDK
• Google ADK (Agent Developer Kit)
• Expertise in tool calling: designing tools, schemas, multi‑agent workflows, and handling structured outputs.
• Strong context engineering skills (retrieval strategies, prompt orchestration, token optimization, guardrails).
• Experience working with or building MCP (Model Context Protocol) tools/servers is a strong plus.
• Good backend fundamentals: SQL/NoSQL, caching, asynchronous processing, and performance debugging.
• Familiarity with system design and production‑ready engineering practices.
• Experience with cloud deployment for AI agents is a plus (AWS/GCP/Azure).
• Strong communication skills and ability to own features end‑to‑end.

Preferred Qualifications

• Experience with cloud platforms (AWS, Azure, GCP) and services relevant to ML (e.g., SageMaker, Vertex AI).
• Familiarity with containerization and orchestration technologies (Docker, Kubernetes).
• Experience in building and maintaining API services (e.g., FastAPI, Flask) to expose ML models for real‑time inference.
• Knowledge of database technologies (SQL and NoSQL).
• Familiarity with data versioning tools (DVC) and model serving frameworks.

Pays and Perks

• Health Insurance for individuals and families
• Encouragement and resources for using cutting‑edge AI tools in your work profile
• Company Sponsorship on relevant professional development courses and certifications
• Flexibility in timing to support work‑life balance
• Friendly and collaborative work environment
• No Micromanagement—focus on autonomy and results
• Great company culture centered on innovation and growth
• Perfect office location for easy access

Company Website: www.attri.ai

Location: Ahmedabad (Google Map)

Job Type: Full‑time

Seniority Level: Entry level

Employment Type: Full‑time

Job Function: Engineering and Information Technology