Skip to Main Content

Job Title


Gen AI Architect ( Deep Learning and Applied Mathematics)


Company : o9 Solutions, Inc.


Location : Pune, Maharashtra


Created : 2026-04-16


Job Type : Full Time


Job Description

Transforming the Future of Enterprise PlanningAt o9, our mission is to be the Most Value-Creating Platform for enterprises by transforming decision-making through our AI-first approach. By integrating siloed planning capabilities and capturing millions—even billions—in value leakage, we help businesses plan smarter and faster.This not only enhances operational efficiency but also reduces waste, leading to better outcomes for both businesses and the planet. Global leaders like Google, PepsiCo, Walmart, T-Mobile, AB InBev, and Starbucks trust o9 to optimize their supply chains.We are seeking a highly experienced, intellectually curious Senior Architect with a strong foundation in Applied Mathematics, Statistical Modeling, and Deep Learning to lead the design, implementation and delivery of next-generation AI models and algorithms.In this role, you will operate at the intersection of business problem-solving, mathematical modeling, software development and Agentic AI development. You will translate complex real-world problems into formal mathematical and statistical representations, design and implement advanced deep learning and Agentic systems, and ensure their reliable deployment in production environments.The ideal candidate combines software engineering, statistical modeling, AI techniques, and a strong passion for applied mathematics and innovation, with the ability to influence technical direction across teams.What you will doArchitectural LeadershipOwn the end-to-end mathematical and statistical architecture of complex AI systems, from data integration and feature engineering to model training, inference, and monitoring.Design scalable, robust and secure AI models and algorithms capable of handling large-scale, complex enterprise systems, grounded in first-principles mathematical design.Establish architectural standards, design principles, and best practices for AI-centric systems across the organization.Applied Mathematics & Statistical ModelingTranslate business and domain problems into mathematical, statistical, and agentic models.Apply foundational techniques from linear algebra, optimization, numerical methods, statistics, causal inference to improve model robustness and performance.Guide teams in selecting appropriate modeling approaches, loss functions, evaluation metrics, and validation strategies.Deep LearningDesign, develop, and deploy advanced deep learning models (e.g., transformers, state-space, diffusion models)Oversee the full model lifecycle, including experimentation, training, evaluation, deployment, and continuous improvement.Ensure models are production-ready, explainable where required, and aligned with business objectives.Model Optimization & Performance EngineeringOptimize models and pipelines for speed, scalability, memory/compute efficiency, and accuracy.Apply techniques such as pruning, quantization, distributed training, and GPU/accelerator optimization.Collaborate with platform and infrastructure teams to maximize system-level performance.Technical Mentorship & InfluenceAct as a technical mentor and thought leader for AI engineers, AI scientists and software developers.Review designs, code, and models, providing guidance on architecture, quality, and maintainability.Elevate modeling maturity by promoting best practices in model design, algorithm development, coding, testing, and documentation.Strategy, Research & InnovationEvaluate, select, and evolve AI frameworks, libraries, tools, and cloud platforms.Stay current with cutting-edge research, industry trends, and emerging technologies in AI and applied mathematics.Drive innovation by identifying opportunities to apply advanced AI techniques to business challenges.What you will haveEducationMasters or PhD in Computer Science, Applied Mathematics, Statistics, Physics, or a closely related quantitative discipline.Professional Experience10+ years of overall software engineering experience.5+ years of hands-on experience in applied mathematics, statistical modeling, and deep learning systems.Demonstrated experience architecting and deploying AI solutions in production environments.Deep Learning ExpertiseStrong experience designing, training, and deploying deep learning models using modern AI frameworks.Solid understanding of the full ML lifecycle, including experimentation, deployment, monitoring, and retraining.Experience working with large datasets and distributed or high-performance computing environments.Mathematics & StatisticsDeep expertise in linear algebra, multivariate calculus, probability theory, optimization, statistical modeling and causal inference.Ability to reason formally about model behavior, convergence, and performance trade-offs.Programming & Engineering SkillsExpert-level proficiency in Python, including state-of-the-art mathematical, statistical, ML and agentic librariesStrong software engineering fundamentals: data structures, algorithms, system design, and performance optimization.Experience with additional languages such as Scala, Kotlin or Rust is highly desirable.More about us… At o9, transparency and open communication are at the core of our culture. Collaboration thrives across all levels—hierarchy, distance, or function never limit innovation or teamwork. Beyond work, we encourage volunteering opportunities, social impact initiatives, and diverse cultural celebrations.With a $3.7 billion valuation and a global presence across Dallas, Amsterdam, Barcelona, Madrid, London, Paris, Tokyo, Seoul, and Munich, o9 is among the fastest-growing technology companies in the world. Through our aim10x vision, we are committed to AI-powered management, driving 10x improvements in enterprise decision-making. Our Enterprise Knowledge Graph enables businesses to anticipate risks, adapt to market shifts, and gain real-time visibility. By automating millions of decisions and reducing manual interventions by up to 90%, we empower enterprises to drive profitable growth, reduce inefficiencies, and create lasting value.o9 is an equal-opportunity employer that values diversity and inclusion. We welcome applicants from all backgrounds, ensuring a fair and unbiased hiring process. Join us as we continue our growth journey!