rippling

Software Engineer - AI Platform (India)

Join us!

Bengaluru, IND

Full-time

Mid · 3+ years experience

Hybrid

Discovered 3 weeks ago

PythonGoTensorFlowPyTorchScikit-LearnFastAPI

Free

Job Fit Check

Base Career helps you apply smarter for this job.

Ready to Scan

Key skills for this role

PythonGoTensorFlow

Build and scale AI/ML and GenAI pipelines from experimental workflows to production-ready systems.
Integrate model training, evaluation, deployment, and monitoring into product workflows
Deploy and manage GenAI solutions such as chatbots, RAG applications, and predictive analytics tools.
Operationalize LLMs and AI agents, including prompt orchestration, chaining, and fine-tuning.
Benchmark models, develop evaluation frameworks, and improve reliability and auditability.
Implement observability, monitoring, and rollback mechanisms to ensure secure, scalable deployments.
Work across the stack—from backend systems to product SDKs—to deliver AI features directly into user-facing applications.
Prototype rapidly, gather feedback, and iterate while keeping scale and maintainability in mind.
Own critical product components and take responsibility for delivering robust, production-grade features.
Collaborate cross-functionally with data scientists, product managers, and engineers to scope specifications and solve real customer problems.
Debug complex issues and perform root cause analysis across model pipelines, infrastructure, and product layers to ensure reliability and continuous improvement.

BS or MS in Computer Science, Statistics, or Mathematics, or equivalent experience.
Strong software engineering background with proven experience shipping production systems.
3+ years of experience in ML/DL pipelines, deployment, and applied AI solutions.
Proficiency in Python or Go with frameworks like TensorFlow, PyTorch, Scikit-Learn, FastAPI, or gRPC.
Experience with LLM and AI frameworks such as Langchain, LlamaIndex, Hugging Face Transformers, and OpenAI API.
Knowledge of RAG architectures, embeddings, reranking models, and LLM-based dialogue systems.
Experience building and scaling backend platforms, APIs, and microservices.

Skip the repetitive application forms

Install the Base Career Chrome Extension and autofill job applications across major job boards with your profile.

Trusted by over 500,000 job seekers on Base Career