Proximity Works Logo

Proximity Works

Backend Engineer - AI/ML (Immediate Joiners only)

Posted 15 Days Ago
Be an Early Applicant
In-Office
Navi Mumbai, Thane, Maharashtra
Senior level
In-Office
Navi Mumbai, Thane, Maharashtra
Senior level
The Senior Backend Engineer will design and operate AI/ML systems, manage backend infrastructure, and collaborate with teams to build production-grade solutions.
The summary above was generated by AI

We are looking for a Senior Backend Engineer to build and operate the core AI/ML-backed systems that power large-scale, consumer-facing products. You will work on production-grade AI runtimes, retrieval systems, and ML-adjacent backend infrastructure, making pragmatic tradeoffs across quality, latency, reliability, and cost.

This role is not an entry point into AI/ML. You are expected to already have hands-on experience shipping ML-backed backend systems in production.

At Proximity, you won’t just build APIs - you’ll own critical backend systems end-to-end, collaborate closely with Applied ML and Product teams, and help define the foundations that power intelligent experiences at scale.

Responsibilities -

  • Own and deliver end-to-end backend systems for AI product runtime, including orchestration, request lifecycle management, state/session handling, and policy enforcement.
  • Design and implement retrieval and memory primitives end-to-end — document ingestion, chunking strategies, embeddings generation, indexing, vector/hybrid search, re-ranking, caching, freshness, and deletion semantics.
  • Productionize ML workflows and interfaces, including feature and metadata services, online/offline parity, model integration contracts, and evaluation instrumentation.
  • Drive performance, reliability, and cost optimization, owning P50/P95 latency, throughput, cache hit rates, token and inference costs, and infrastructure efficiency.
  • Build observability by default, including structured logs, metrics, distributed tracing, guardrail signals, failure taxonomies, and reliable fallback paths.
  • Collaborate closely with Applied ML teams on model routing, prompt and tool schemas, evaluation datasets, and release safety gates.
  • Write clean, testable, and maintainable backend code, contributing to design reviews, code reviews, and operational best practices.
  • Take systems from design → build → deploy → operate, including on-call ownership and incident response.
  • Continuously identify bottlenecks and failure modes in AI-backed systems and proactively improve system robustness.

Requirements
  • Bachelor’s degree in Computer Science, Engineering, or a related field, or equivalent practical experience.
  • 6–10 years of experience building backend systems in production, with 2–3+ years working on ML/AI-backed products such as search, recommendations, ranking, RAG pipelines, or AI assistants.
  • Strong practical understanding of ML system fundamentals, including embeddings, vector similarity, reranking, retrieval quality, and evaluation metrics (precision/recall, nDCG, MRR).
  • Proven experience implementing or operating RAG pipelines, covering ingestion, chunking, indexing, query understanding, hybrid retrieval, and rerankers.
  • Solid distributed systems fundamentals, including API design, idempotency, concurrency, retries, circuit breakers, rate limiting, and multi-tenant reliability.
  • Experience with common ML/AI platform components, such as feature stores, metadata systems, streaming or batch pipelines, offline evaluation jobs, and A/B measurement hooks.
  • Strong proficiency in backend programming languages and frameworks (e.g., Go, Java, Python, or similar) and API development.
  • Ability to work independently, take ownership of complex systems, and collaborate effectively with cross-functional teams.
  • Strong problem-solving, communication, and system-design skills.
    Desired Skills -
  • Experience with agentic runtimes, including tool-calling or function-calling patterns, structured outputs, and production guardrails.
  • Hands-on exposure to vector and hybrid retrieval stacks such as FAISS, Milvus, Pinecone, or Elasticsearch.
  • Experience running systems on Kubernetes, with strong knowledge of observability stacks like OpenTelemetry, Prometheus, Grafana, and distributed tracing.
  • Familiarity with privacy, security, and data governance considerations for user and model data.

Benefits
  • Best in class compensation: We hire only the best, and we pay accordingly.
  • Proximity Talks: Meet engineers, designers, and product leaders — and learn from experts across domains.

Keep on learning with a world-class team: Work on real, production AI systems at scale, challenge yourself daily, and grow alongside some of the best minds in the industry.

About us -

We are Proximity — a global team of coders, designers, product managers, geeks, and experts. We solve complex problems and build cutting-edge technology at scale.

Our team of Proxonauts is growing quickly, which means your impact on the company’s success will be significant. You’ll work with experienced leaders who have built and led multiple high-performing tech, product, and design teams.

Here’s a quick guide to getting to know us better:

  • Watch our CEO, Hardik Jagda, tell you all about Proximity.
  • Read about Proximity’s values and meet some of our Proxonauts.
  • Explore our website, blog, and design wing — Studio Proximity.
  • Get behind the scenes with us on Instagram — follow @ProxWrks and @H.Jagda.

Top Skills

AI
Api Development
Elasticsearch
Faiss
Go
Grafana
Java
Kubernetes
Milvus
Ml
Opentelemetry
Pinecone
Prometheus
Python

Similar Jobs

51 Minutes Ago
Easy Apply
Hybrid
Mumbai, Maharashtra, IND
Easy Apply
Senior level
Senior level
Artificial Intelligence • Consumer Web • Edtech • Enterprise Web • HR Tech • Social Impact • Generative AI
The Growth Enterprise Account Executive will manage the sales cycle, build client relationships, develop strategies, and collaborate with teams to drive revenue and expand client accounts.
Top Skills: B2B Saas
2 Hours Ago
Hybrid
Mumbai, Maharashtra, IND
Senior level
Senior level
Digital Media • eCommerce • Gaming • Mobile • News + Entertainment
Lead localized lifecycle marketing strategies for India, focusing on subscriber engagement across acquisition, onboarding, retention, and winback campaigns.
Top Skills: BrazeSmsUpiWhatsapp
5 Hours Ago
Hybrid
Pune, Maharashtra, IND
Mid level
Mid level
Artificial Intelligence • Cloud • Internet of Things • Machine Learning • Analytics • Industrial
The Lead Software Engineer will develop solutions for complex challenges at John Deere, leveraging advanced technology and innovation.

What you need to know about the Mumbai Tech Scene

From haggling for the best price at Chor Bazaar to the bustle of Crawford Market, the energy of Mumbai's traditional markets is a key part of the city's charm. And while these markets will always have their place, the city also boasts a thriving e-commerce scene, ranking among the largest in the region. Driven by online sales in everything from snacks to licensed sports merchandise to children's apparel, the local industry is worth billions, with companies actively recruiting to meet the demands of continued growth.

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account