Senior Technical Lead / Senior Solutions Architect (AI/ML)
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.
qualification–
responsibilities—
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.
business partnership related—
supporting tasks will include—
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.
role relationships—
technical skills—
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.
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—
- The Proximity Foundation is dedicated to enriching the lives of underprivileged and high-risk groups of women and children through accessible education.
- Our goal is to provide them with a well-rounded education that combines academics with the essential life skills necessary to thrive in the 21st century and beyond.
- In a heartfelt post, our Founder and CEO, Hardik, shares his personal journey and the inspiration behind the Proximity Foundation. You can watch it here.
- Visit our website to discover our innovative programs and initiatives and learn how you can be a part of our mission by clicking here.
- Follow us on social media handles for other stories and updates:
Instagram
LinkedIn
X (formerly Twitter)
about us—
Proximity is the trusted technology, design, and consulting partner for leading startups, fast-growing scale-ups, and global enterprises. We’re headquartered in San Francisco and have offices in Palo Alto, Dubai, Mumbai, and Bangalore. Since 2019, Proximity has created and grown high-impact, scalable products used by 370 million daily users, with a total net worth of $45.7 billion among our client companies.
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
To apply, email us a one-pager writeup (including links to your work) and your resume at founders@proximity.tech
