Senior Applied Scientist - AI Agent Systems

Auxo AI; Gurugram District; Hybrid; INR 4,000,000 - 7,000,000; Full time

Job summary

Auxo AI is seeking a Senior Applied Scientist to design and deploy production-grade AI agents capable of structured reasoning, planning, and decision-making. The role emphasizes planning, search, and optimization over chatbot work, with hybrid in-office requirements in Mumbai/Bangalore/Hyderabad/Gurgaon.

The candidate will architect modular AI agents, implement MCTS and graph-based planning, and build memory and tool-calling architectures for reliable real-world use.

Qualifications

• Must have experience designing AI systems in production environments.
• Strong grasp of planning or search-based decision making beyond basic cases.
• Hands-on experience with MCTS or related decision-making frameworks.
• Understanding of state-space representations and heuristic design.
• Experience building agent frameworks for real-world applications.
• Experience designing tool-use architectures under practical constraints.
• Familiarity with performance optimization for latency, cost, and reliability.

Responsibilities

• Design modular AI agent frameworks with skill decomposition, tool orchestration, and persistent state tracking.
• Implement planning and search algorithms (MCTS, beam search, A*, heuristic search) and graph-based planning for complex decisions.
• Develop decision-making loops balancing exploration vs exploitation, cost vs accuracy, and latency vs reasoning depth.
• Build structured memory systems: episodic, semantic, and vector memories with efficient retrieval.
• Design tool-calling architectures with execution validation, retries, and recovery strategies.
• Develop evaluation frameworks using task success metrics, rollout simulations, and multi-sample validation.
• Improve agent performance via distillation, synthetic trajectory generation, prompt compression, and context pruning.
• Deliver production-ready agent systems meeting reliability, cost, throughput, and observability.

Skills

Planning algorithms

Monte Carlo Tree Search

Python

Agent frameworks

Tool orchestration

Memory systems

Graph algorithms

Latency optimization

Reinforcement learning

Job description

AuxoAI is hiring a Senior Applied Scientist to design and deploy production-grade AI agents capable of structured reasoning, planning, and decision-making.

This role focuses on building reasoning and decision systems using planning algorithms, search methods, and optimization techniques, rather than chatbot or RAG-style application development. The ideal candidate will design intelligent agent architectures that combine LLM-based reasoning with classical planning, search algorithms, and optimization techniques, operating reliably in real-world environments with constraints around latency, cost, uncertainty, and limited context windows.

You will work on advanced AI systems that power autonomous workflows, decision engines, and tool-driven agent ecosystems.

You will also work on problems where existing architectures may not be sufficient, and will be expected to experiment with new approaches that combine machine learning, graph algorithms, and classical AI techniques to build reliable, production-grade systems.

Location - Mumbai/Bangalore/Hyderabad/Gurgaon (Hybrid - 3 Days a week in Office)

Responsibilities

• Design and architect modular AI agent frameworks incorporating skill decomposition, tool orchestration, and persistent state tracking.


• Implement planning and search algorithms such as Monte Carlo Tree Search (MCTS), beam search, A search, heuristic search, and graph-based planning approaches* to support complex decision-making tasks.


• Develop decision-making loops that balance trade-offs between exploration vs. exploitation, cost vs. accuracy, and latency vs. reasoning depth.


• Build structured memory systems including episodic memory stores, semantic memory layers, and vector-based memory with optimized retrieval strategies.


• Design tool-calling architectures with strong execution validation, retry mechanisms, and failure recovery strategies.


• Develop evaluation frameworks to measure agent performance using task success metrics, rollout simulations, and multi-sample validation approaches.


• Improve agent performance through techniques such as distillation, synthetic trajectory generation, prompt compression, and context pruning.


• Deliver production-ready agent systems that meet operational requirements around reliability, cost efficiency, throughput, and observability.

Requirements

• 3-10 years of experience building machine learning or AI systems in production environments.


• Strong experience implementing search or planning algorithms beyond basic use cases, including tree search or heuristic-based planning approaches.


• Hands‑on experience with Monte Carlo Tree Search (MCTS) or related decision‑making frameworks.


• Strong understanding of state‑space representations, heuristic design, and decision boundary trade‑offs.


• Experience building or extensively customizing agent frameworks for real‑world applications.


• Hands‑on experience designing tool‑use or function‑calling architectures under practical system constraints.


• Strong Python engineering skills with a focus on scalable and reliable system design.

Candidates whose primary experience is limited to RAG pipelines, prompt engineering, or chatbot frameworks without deeper algorithmic or systems work may not be a fit for this role.

Nice To Have

• Experience with reinforcement learning techniques such as policy gradients, value estimation, or reward modeling.


• Experience building multi‑agent or collaborative agent systems.


• Experience designing evaluation frameworks for agent robustness and reliability.


• Experience optimizing LLM inference pipelines for latency, throughput, and cost efficiency.


• Familiarity with distributed task orchestration systems and large‑scale AI workflow management.

Skills:- Machine Learning (ML), Retrieval Augmented Generation (RAG) and Agentic AI