Senior Technical Lead / Senior Solutions Architect (AI/ML)

Job Description

We are looking for a Senior Technical Lead / Senior Solutions Architect to design, develop, and scale innovative AI/ML-driven solutions. You will be responsible for architecting highly scalable, low-latency distributed systems optimized for AI/ML workloads. As a key technical leader, you will solve complex challenges, influence next-generation AI/ML infrastructures, and guide cross-functional teams to deliver state-of-the-art solutions for fast-growing startups and enterprise companies.

Be at the forefront of shaping next-generation AI/ML infrastructures, driving solutions for high-impact products across diverse industries. You’ll have the opportunity to influence key architectural decisions and enable real-world applications that scale globally, ensuring innovation and efficiency at every step.

Requirements

You’ll be responsible for —

• Lead the execution of large-scale AI/ML projects, ensuring alignment with business objectives.
• Architect and design scalable, low-latency backend architectures, focusing on AI/ML model serving, real-time data processing, and integration with cloud infrastructure.
• Oversee deployment of AI/ML models and infrastructure, ensuring high-performance, reliability, and scalability in production environments.
• Drive decisions for cloud platforms (AWS, GCP, or Azure), auto-scaling, serverless architectures, and resource optimization.
• Influence and guide the architectural direction of the organization, ensuring future-proof, AI/ML-ready systems and solutions.
• Define benchmarks and optimize performance across AI/ML workloads.
• Lead technical troubleshooting, performance tuning, and architecture optimizations to ensure scalable and efficient deployment of AI/ML systems.
• Mentor and provide technical guidance to engineers across multiple teams, fostering a collaborative and innovative engineering culture.
• Communicate project progress, risks, and technical decisions effectively to stakeholders, ensuring transparency and alignment with business goals.
• Ensure engineering best practices are followed, including CI/CD, TDD, version control, and test automation.

You need —

• 10+ years in software engineering or architecture, including 5+ years leading cross-functional teams in AI/ML or distributed systems projects.
• Proven experience designing and managing large-scale backend architectures and distributed environments, with expertise in microservices, event-driven systems, and RESTful/GraphQL API development for AI/ML systems in production.
• Extensive experience with AI/ML model deployment, performance tuning, model serving, and seamless integration with backend systems.
• Deep understanding of cloud platforms (AWS, GCP, Azure), auto-scaling, serverless, and fault-tolerant architectures.
• Strong experience in machine learning frameworks (TensorFlow, PyTorch) and data pipelines for real-time processing and high-concurrency environments.
• Ability to troubleshoot and resolve complex technical challenges related to AI/ML workloads, scalability, and performance.
• Excellent leadership and communication skills, with a proven ability to mentor engineers and work closely with cross-functional stakeholders.
• Self-starter who excels at execution, balancing short-term technical delivery with long-term scalability and efficiency goals.

Bonus points for —

• Experience with AI/ML infrastructure tools like MLFlow, Kubeflow, or MLOps best practices.
• Experience with real-time data processing & streaming technologies (Apache Kafka, Flink, Spark).
• Familiarity with NLP frameworks (e.g., Hugging Face, spaCy)
• Open-source contributions, side projects, or thought leadership through technical content (blogs, videos).
• Knowledge of data privacy and security regulations relevant to AI/ML deployments.

Benefits

what you get –

• Best in class salary: We hire only the best, and we pay accordingly.
• Proximity Talks: Meet other designers, engineers, and product geeks — and learn from experts in the field.
• Keep on learning with a world-class team: Work with the best in the field, challenge yourself constantly, and learn something new every day.

about us –

We are Proximity — a global team of coders, designers, product managers, geeks, and experts. We solve complex problems and build cutting-edge tech, at scale. Our team of Proxonauts is growing quickly, which means your impact on the company’s success will be huge. You’ll have the chance to work with experienced leaders who have built and led multiple 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 here.
• Explore our website, blog, and the design wing — Studio Proximity.
• Get behind-the-scenes with us on Instagram! Follow @ProxWrks and @H.Jagda.