Deep Learning Compiler Engineer Jobs

Secure Every Identity, from AI to Human Identity is the key to unlocking the potential of AI. Okta secures AI by building the trusted, neutral infrastructure that enables organizations to safely embrace this new era. This work requires a relentless drive to solve complex challenges with real-world stakes. We are looking for builders and owners who operate with speed and urgency and execute with excellence. This is an opportunity to do career-defining work. We're all in on this mission. If you are too, let's talk. Okta is looking for a results-oriented and strategic Head of Marketing who loves driving revenue and digging into new Go-To-Market challenges. You will lead the team of marketing professionals and lead the creation of the plan of programs / regionalized campaigns targeting all segments in the key geographies to fuel our APJ expansion. You are a seasoned growth marketer who thrives in a dynamic environment and moves quickly to build structure and process. You will be comfortable collaborating across the entire APJ leadership team, leading the development of a comprehensive marketing plan, and managing a team of 10+ spanning regional marketing, partner marketing, and customer growth. You will also serve as the primary liaison with global teams, who are based across our corporate offices and remote locations worldwide. DUTIES AND RESPONSIBILITIES: Drive APJ Marketing Strategy and Pipeline Generation: Define and own the comprehensive APJ demand marketing strategy, covering all segments from SMB to the region’s largest accounts, and maximizing total pipeline by integrating deep Business/Sales Development knowledge. Maximize Customer Growth: Define and execute the post-sales marketing strategy, focusing on increasing customer retention and adoption, and accelerating expansion through targeted upsell, cross-sell, advocacy and loyalty programs, including user group initiatives. Foster Executive Engagement & Regional Event Strategy: Own and execute the reg

Who we are About Stripe Stripe is a financial infrastructure platform for businesses. Millions of companies—from the world’s largest enterprises to the most ambitious startups—use Stripe to accept payments, grow their revenue, and accelerate new business opportunities. Our mission is to increase the GDP of the internet, and we have a staggering amount of work ahead. That means you have an unprecedented opportunity to put the global economy within everyone’s reach while doing the most important work of your career. About the team Finance and Strategy Data Science builds the forecasting models, data infrastructure, and analytics tools at the core of how Stripe measures and plans its business. The team owns everything from hierarchical time series and agentic forecasting tools that predict payment volumes and revenue margins, to the governed metrics platform that feeds company-wide dashboards and executive reporting. We partner closely with Finance and Strategy, GTM, and Product stakeholders to directly inform financial decisions across Stripe's entire business. The team combines technical depth, strategic thinking, and executive partnership that develops both technical and business expertise. What you’ll do Data Science Managers at Stripe are responsible for the success of their team. You'll be deeply involved in the modeling and design processes as well as coaching, mentoring, and leading the team. You'll have a deep understanding of how to drive efficient data science teams and you'll have a strong user-focus. You'll be working with data scientists, analysts, and engineers on creating technical solutions and communicating effectively across teams and senior leadership. Responsibilities Drive the roadmap and priorities for your team, and work with many Stripe leaders across the company to enhance our ability to be data-driven. Collaborate with stakeholders across the organization such as engineering, analytics, operations, finance, and marketing. Lead and manage proc

At Bolna, we’re building tools that change the way teams leverage Voice AI. We’re looking for a Founding Machine Learning Engineer to own the end-to-end lifecycle of building, evaluating, deploying, and improving models that power millions of production conversations. This is a high-impact, high ownership role where you won’t just work on Bolna’s ML stack—you’ll help build the foundation it scales on. Our team includes IIT alumni with experience at Bain, Atlassian, Uber, Zomato, and LinkedIn, and is backed by leading investors. Responsibilities: Build the data engine - Design pipelines to source and clean conversational voice data across Indian languages, accents, and telephony conditions. Fine-tune models that ship - Fine tune and train models to improve accuracy, speed, and reliability across different use-cases. Define what "good" means - Build evaluation datasets and benchmarks for transcription accuracy, voice naturalness, interruption handling, latency, and end-to-end conversation quality. Set up human-in-the-loop pipelines to capture subjective quality at scale. Ship to production - Work with the engineering team to deploy models into a latency-sensitive, high-volume system. Monitor performance in the wild, debug regressions, and iterate fast. Required Skills: 3+ years of hands-on ML experience with deep practical real-world experience in training models. Strong Python and PyTorch fundamentals with exposure in distributed training, and modern fine-tuning techniques (LoRA, QLoRA, DPO, RLHF, etc.). Training data as a first-class problem. Experience designing data pipelines from collection, cleaning, labeling, deduplication, augmentation and treating data quality as a core engineering discipline. Rigorous about evaluation. You know that "looks good in a demo" is not a benchmark. You build the evals before you trust the model. Speech model experience is a plus with real-time / streaming inference experience where you would have contributed to latency optimization

About the Role We are looking for a senior partner success and learning professional to activate cloud partners to co-sell and deploy customer solutions at scale. This role leads the enablement strategy and execution that equip partner teams to turn joint priorities into measurable customer and business outcomes. As a trusted advisor to strategic partners, you will turn shared business priorities into practical enablement programs that help partner sales, technical, customer success, and leadership teams identify joint opportunities, co-sell effectively, and deploy customer solutions successfully. You will own the partner enablement relationship for an assigned portfolio and work across internal teams to ensure programs are coordinated, scalable, and connected to measurable partner, customer, and business outcomes. This is not a traditional training delivery role or a quota-carrying partner account manager position. It is a consultative, partner-facing role for someone who can shape strategy, build relationships, translate complex AI capabilities into effective learning experiences, lead persuasive enablement sessions, and guide strategic partners toward sustained success. In this role, you will: Own strategic partner enablement and success Serve as the primary enablement and success lead for a portfolio of strategic cloud platforms. Develop a deep understanding of each partner’s business, priorities, audiences, existing learning infrastructure, and strategic relationship with our company. Build partner-specific enablement strategies aligned to shared goals, priorities, and the broader partnership strategy. Establish trusted relationships with partner leaders, alliance teams, enablement stakeholders, sales organizations, technical communities, and post-sales teams. Lead ongoing planning conversations, enablement reviews, stakeholder check-ins, and executive updates that keep partners and internal teams aligned. Identify gaps, risks, opportunities, and emerging partn

About the Team Training Runtime designs the core distributed machine-learning training runtime that powers everything from early research experiments to frontier-scale model runs. With a dual mandate to accelerate researchers and enable frontier scale, we’re building a unified, modular runtime that meets researchers where they are and moves with them up the scaling curve. Our work focuses on three pillars: high-performance, asynchronous, zero-copy tensor and optimizer-state-aware data movement; performant, high-uptime, fault-tolerant training frameworks (training loop, state management, resilient checkpointing, deterministic orchestration, and observability); and distributed process management for long-lived, job-specific and user-provided processes. We integrate proven large-scale capabilities into a composable, developer-facing runtime so teams can iterate quickly and run reliably at any scale, partnering closely with model-stack, research, and platform teams. Success for us is measured by raising both training throughput (how fast models train) and researcher throughput (how fast ideas become experiments and products). About the Role As a Training: ML Framework Engineer, you will work on improving the training throughput for our internal training framework, while enabling researchers to experiment with new ideas. This requires good engineering (for example designing, implementing, and optimizing state-of-the-art AI models), writing bug-free machine learning code (surprisingly difficult!), and acquiring deep knowledge of the performance of supercomputers. In all the projects this role pursues, the ultimate goal is to push the field forward. We’re looking for people who love optimizing performance, understanding distributed systems, and who cannot stand having bugs in their code. Since our training framework is used for large runs with massive numbers of GPUs, performance improvements here will have a large impact. This role is based in San Francisco, CA. We use a

About the Company: Sigmoid enables business transformation using data and analytics, leveraging real-time insights to make accurate and fast business decisions, by building modern data architectures using cloud and open source. Some of the world’s largest data producers engage with Sigmoid to solve complex business problems. Sigmoid brings deep expertise in data engineering, predictive analytics, artificial intelligence, and DataOps. Sigmoid has been recognized as one of the fastest growing technology companies in North America, 2021, by Financial Times, Inc. 5000, and Deloitte Technology Fast 500. Offices : New York | Dallas | San Francisco | Lima | Bengaluru The below role is for our Bengaluru office. Why Join Sigmoid? Sigmoid provides the opportunity to push the boundaries of what is possible by seamlessly combining technical expertise and creativity to tackle intrinsically complex business problems and convert them into straight-forward data solutions. Despite being continuously challenged, you are not alone. You will be part of a fast-paced diverse environment as a member of a high-performing team that works together to energize and inspire each other by challenging the status quo. Vibrant inclusive culture of mutual respect and fun through both work and play. About the Role : Sigmoid is looking to onboard a Senior Leader for our FS Consulting Practice who can bridge the gap between high-level strategy and granular operational reality. A true expert in the Financial services space must understand the interconnectedness of Financial Products & Market Dynamics. In this role, you will drive strategic initiatives, foster innovation, and provide visionary direction to our BFSI practice. The role requires a strong portfolio leader responsible for managing 3–6 key accounts across a mix of new business development (hunting) and account growth (farming). The individual will typically oversee a team of approximately 15–70 mem

About the Team The Technical Success team helps OpenAI’s customers realize meaningful and sustained value from our technology. We partner with customers throughout their journey—from initial exploration and solution design to production implementation and organization-wide adoption. Applied AI Engineers serve as trusted technical partners to customer executives, engineering teams, product leaders, security organizations, and transformation teams. They combine deep technical judgment with strong customer instincts, translating frontier AI capabilities into secure, reliable systems and durable business outcomes. About the Role We are seeking a Manager to build, lead, and develop a high-performing team of Applied AI Engineers supporting our enterprise customers. You will be accountable for the technical success of a broad and strategically important customer portfolio. You will help your team identify high-value opportunities, design and deploy production-grade AI systems, navigate complex technical and organizational constraints, and expand successful implementations across workflows, teams, and business units. This role requires technical depth, people leadership, customer judgment, and operational rigor. You should be comfortable coaching engineers through architecture and evaluation decisions, engaging directly in high-stakes customer situations, and collaborating with Sales, Solutions Engineering, Product, Research, Engineering, Security, and Legal. You will also help define how we serve enterprise customers at scale by developing effective coverage models, reusable implementation patterns, technical enablement, escalation mechanisms, and systems for turning field insights into high-quality product feedback. This role is open in our Singapore office. We use a hybrid work model of 3 days in the office per week and offer relocation assistance to new employees. In This Role, You Will Build, manage, and develop a high-performing team of Applied AI Engineers supporting

About the Team The Technical Success team helps OpenAI’s customers realize meaningful and sustained value from our technology. We partner with customers throughout their journey—from initial exploration and solution design to production implementation and organization-wide adoption. Applied AI Engineers serve as trusted technical partners to customer executives, engineering teams, product leaders, security organizations, and transformation teams. They combine deep technical judgment with strong customer instincts, translating frontier AI capabilities into secure, reliable systems and durable business outcomes. About the Role We are seeking a Manager to build, lead, and develop a high-performing team of Applied AI Engineers supporting our enterprise customers. You will be accountable for the technical success of a broad and strategically important customer portfolio. You will help your team identify high-value opportunities, design and deploy production-grade AI systems, navigate complex technical and organizational constraints, and expand successful implementations across workflows, teams, and business units. This role requires technical depth, people leadership, customer judgment, and operational rigor. You should be comfortable coaching engineers through architecture and evaluation decisions, engaging directly in high-stakes customer situations, and collaborating with Sales, Solutions Engineering, Product, Research, Engineering, Security, and Legal. You will also help define how we serve enterprise customers at scale by developing effective coverage models, reusable implementation patterns, technical enablement, escalation mechanisms, and systems for turning field insights into high-quality product feedback. In This Role, You Will Build, manage, and develop a high-performing team of Applied AI Engineers supporting large and complex enterprise customers. Own the quality and impact of the team’s work across solution design, implementation, production readiness, adop

Fiduciary Trust International is a premier investment and wealth management firm with a commitment to growing and protecting wealth across generations. We offer a dynamic and collaborative approach to managing wealth for high-net-worth and ultra high-net-worth individuals and families, family offices, endowments, foundations, and institutions. Our investment managers, tax and estate planning professionals work together to develop holistic strategies to optimize clients’ portfolios while mitigating the impact of taxes on their wealth. As a fiduciary, the guidance we provide is in the best interests of our clients, without conflict or competing benefits. We offer boutique customization and deep expertise in specialized investment, tax and planning strategies alongside sophisticated technology and custody platforms. Fiduciary Trust International is owned by Franklin Templeton, a dynamic firm that spans asset management, wealth management, and fintech, giving us many ways to help investors make progress toward their goals. With clients in over 150 countries and offices on six continents, you’ll get exposed to different cultures, people, and business development happening around the world. About the Department: Our AI Engineering team develops and scales production-ready Generative AI and Agentic AI solutions that help solve complex business challenges and accelerate innovation across the organization. The team brings together AI engineers, architects, and business stakeholders who work collaboratively to design, build, and deploy secure, enterprise-grade solutions on Microsoft Azure. Joining this team offers opportunities to work with emerging technologies, influence AI standards and best practices, and contribute to high-impact initiatives in a collaborative and growth-oriented environment. How Y

We are looking for a Solutions Architect to help customers and partners in South East Asia embrace NVIDIA technologies to build, deploy, and scale vision AI and video AI solutions. This is a highly technical role that requires deep expertise in VLM AI models and software engineering practices. We need a passionate, hard-working, expert and creative individual to help us pursue the many opportunities in this region. A Solution Architect is the first line of technical expertise between NVIDIA and our customers, as well as our partners. Your duties will vary from solutions design, training/workshops, troubleshooting, project coordination, industry and marketing speaking engagements, customer relationship management and more. What you'll be doing: Drive the adoption of NVIDIA's vision AI and video AI capabilities. Provide hands-on technical leadership to put VLM (Visual Language Models) and WFM (World Foundation Models) into applications for computer vision, document processing, and video analytics. Support solution development and technical validation around NVIDIA vision AI and video AI technologies Work with field, product, and engineering teams to translate customer requirements into deployable architectures, technical feedback, and roadmap input. Deliver demos, workshops, technical reviews, and partner enablement sessions that accelerate adoption of NVIDIA AI technologies. What we need to see: BS, MS, or PhD in Computer Science, Electrical Engineering, Computer Engineering, Physics, Mathematics, Machine Learning, or a related field, or equivalent experience. 5+ years of proven experience in computer vision, vision AI, video AI, or world foundation models. Hands-on experience in building and scaling production VLM and multi-modal AI systems. Strong Python ski

ABOUT BASETEN Baseten powers mission-critical inference for the world's most dynamic AI companies, like Cursor, Notion, OpenEvidence, Abridge, Clay, Gamma and Writer. By uniting applied AI research, flexible infrastructure, and seamless developer tooling, we enable companies operating at the frontier of AI to bring cutting-edge models into production. We're growing quickly and recently raised our $1.5B Series F , led by Altimeter Capital, Conviction Partners, and Spark Capital. Join us and help build the platform engineers turn to to ship AI products. THE ROLE Are you passionate about advancing the application of artificial intelligence? We are looking for a Software Engineer focused on ML performance to join our dynamic team. This role is ideal for someone who thrives in a fast-paced startup environment and is eager to make significant contributions to the exciting field of LLM Inference. If you are a backend engineer who thrives on making things faster and is excited about open-source ML models, we look forward to your application. EXAMPLE INITIATIVES You'll get to work on these types of projects as part of our Model Performance team: Baseten Embeddings Inference: The fastest embeddings solution available The Baseten Inference Stack Driving model performance optimization RESPONSIBILITIES Implement, refine, and productionize cutting-edge techniques (quantization, speculative decoding, kv cache reuse, chunked prefill and LoRA) for ML model inference and infrastructure. Deep dive into underlying codebases of TensorRT, PyTorch, TensorRT-LLM, vllm, sglang, CUDA, and other libraries to debug ML performance issues. Apply and scale optimization techniques across a wide range of ML models, particularly large language models. Collaborate with a diverse team to design and implement innovative solutions. Own projects from idea to production. REQUIREMENTS Bachelor's, Master's, or Ph.D. degree in Computer Science, Engineering, Mathematics, or related field. Experience with one

Who are we? Cohere is the leading security-first enterprise AI company. We build cutting-edge foundation AI models and end-to-end products that are designed to solve real-world business problems. We’re training and deploying frontier models for enterprises who are building AI systems. We believe that our work is instrumental to the widespread adoption of AI and we are looking for folks that want to be part of that. We obsess over what we build. Each one of us is responsible for contributing to increasing the capabilities of our models and the value they drive for our customers. Cohere is a team of researchers, engineers, designers, and more, who are all passionate about their craft. We are a global technology company headquartered in Toronto with key offices in London, New York City, San Francisco, Montreal, Paris, Berlin and Seoul. Join us! Why this role? To have the opportunity to collaborate with Cohere researchers and tools on designing and implementing novel research ideas and shipping state-of-the-art models to production. We have openings in teams covering base model training, retrieval augmented generation, data and evaluation, safety, and finetuning, to name a few; and we are open to receiving intern applications in any research area relating to LLMs to broaden your research connections while obtaining deep experience in a growing AI startup. Please Note: To be eligible for a Research Internship, you must be currently pursuing a PhD in Machine Learning, NLP, or a related discipline. You need to be available for a full-time internship that lasts for 4-6 months. As a Cohere Research Intern, you will: Conduct cutting-edge machine learning research, building and training large language models. Focus on research projects aimed at expanding the frontier of knowledge in language modelling and associate areas such as evaluation, multimodal models, optimisation etc. Disseminate your research results through the production of publications, datasets, and code. Contrib

At ClickUp, we're building the future of work: the first truly converged AI workspace unifying tasks, docs, chat, calendar, and enterprise search, all supercharged by context-driven AI. We are an AI-native company. Every team member is expected to leverage AI daily, and we evaluate AI fluency as part of our hiring process. Join us and help redefine what's possible. 🚀 We are seeking a highly technical and experienced Engineering Manager to one of our business critical engineering team. This role is ideal for a hands-on leader who thrives in a fast-paced environment, has a deep understanding of collaborative document editing technologies, and is passionate about building scalable, high-performance software. As the Manager, you will oversee the development and delivery of innovative features, ensure technical excellence, and mentor a team of talented engineers. The Role: Technical Leadership : Provide hands-on technical guidance to the team, ensuring best practices in software development, architecture, and design. Team Management : Lead, mentor, and grow a team of engineers, fostering a culture of collaboration, innovation, and continuous improvement. Product Development : Drive the development of new features and enhancements, ensuring high performance, scalability, and reliability. Collaboration : Work closely with product managers, designers, and other engineering teams to align on goals, prioritize initiatives, and deliver exceptional user experiences. Code Quality : Oversee code reviews, ensure adherence to coding standards, and advocate for clean, maintainable, and testable code. Innovation : Stay up-to-date with the latest trends and technologies in collaborative editing, cloud infrastructure, and web development, and apply them to improve our product. Operational Excellence : Ensure the stability and performance of the Docs platform, proactively addressing technical debt and optimizing system architecture. Qualifications: Technical Expertise : Proficiency in

Secure Every Identity, from AI to Human Identity is the key to unlocking the potential of AI. Okta secures AI by building the trusted, neutral infrastructure that enables organizations to safely embrace this new era. This work requires a relentless drive to solve complex challenges with real-world stakes. We are looking for builders and owners who operate with speed and urgency and execute with excellence. This is an opportunity to do career-defining work. We're all in on this mission. If you are too, let's talk. Okta: The World’s Identity Company™ About Okta & The TAM Team Okta is the leader in Identity management and The World’s Identity Company™. Our mission is to build a more secure world by providing a neutral, powerful, and extensible platform that puts identity at the heart of the modern tech stack—securing customers, employees, partners, and AI agents. Two-thirds of the Fortune 100 trust Okta to power their identity journeys. The Technical Account Management (TAM) team is a global group of Okta product experts dedicated to driving long-term customer success. As a Senior TAM within our Strategic Workforce Identity segment, you will serve as the premier technical advisor and advocate for some of our largest enterprise customers (across Financial Services, Healthcare, Public Sector, and more), helping them elevate their technical maturity and maximize their Okta investment. What You’ll Be Doing 1. Strategic Customer Partnership & Influence Trusted Advisor: Build deep, executive-level relationships, serving as the primary technical point of contact for customer identity owners and IT leadership (up to CxO). Strategic Roadmapping: Architect tailored technical success plans aligned with customer business objectives to drive platform adoption and secure Okta against competitive displacement. Adoption & Expansion: Guide customers in expanding their Okta footprint across critical Workforce solutions, continuously improving their overall security po

Secure Every Identity, from AI to Human Identity is the key to unlocking the potential of AI. Okta secures AI by building the trusted, neutral infrastructure that enables organizations to safely embrace this new era. This work requires a relentless drive to solve complex challenges with real-world stakes. We are looking for builders and owners who operate with speed and urgency and execute with excellence. This is an opportunity to do career-defining work. We're all in on this mission. If you are too, let's talk. About Okta for AI Agents Okta secures access for 20,000 organizations and billions of users. Okta for AI Agents extends that work to the agentic shift. Deploying an AI agent is not like deploying traditional software. You are putting professional work output into production, and it needs deep integration, continuous tuning, and change management. Every agent needs an identity, a scope, an audit trail, and a way to be shut down when it goes wrong. Most enterprises have not built this yet. We are. We hire builders who see the cracks in enterprise agent identity that everyone else has learned to live with. The Role You are the most senior technical field authority for agent identity at Okta. Where a Senior FDE owns the outcome inside one account, you own the patterns that every account and every FDE inherits. You take the hardest and most strategic deployments yourself, set the reference architecture the team builds from, and turn what the field learns into the direction the product takes. You still write code. You also multiply the people around you, and you are the person product and engineering leadership call when an agent identity problem has no precedent. Responsibilities Own the reference architecture. Define the canonical agent identity, delegation, audit, and kill-switch patterns that Senior FDEs deploy across the portfolio, and keep them current as the standards and the product move. Lead the hardest accounts. Personally own the most strategic, regul