Ai Ml Infra Systems Engineer Jobs

Staff Software Engineer - AI Research Infrastructure P-1215 At Databricks, we are obsessed with enabling data teams to solve the world’s toughest problems, from security threat detection to cancer drug development. We do this by building and running the world’s best data and AI platform so our customers can focus on the high-value challenges that are central to their own missions. The Databricks AI Research organization enables companies to develop AI models and agents using their own data, with technologies ranging from post-training open source LLMs to developing advanced multi-agent architectures. Databricks AI does so by producing novel science and putting it into production. Databricks AI is committed to the belief that a company’s AI models and agents are just as valuable as any other core IP, and that high-quality AI should be available to all. Job Description As a Staff Software Engineer, AI Research Infrastructure, you will be developing and running the research stack that powers Databricks AI Research. You will design and build services that schedule, orchestrate, and observe large‑scale training and inference experiment workloads across thousands of GPUs, improve our dev tooling and ensure that researchers can iterate quickly without sacrificing reliability, efficiency, or security. You’ll partner closely with research scientists, ML engineers, and platform teams to turn experimental workloads into robust, repeatable pipelines, and to push the limits of what our infrastructure can support. The Impact you will have As a Staff Software Engineer on the AI Research Infra Team at Databricks, you will: Design and implement infrastructure that supports large‑scale experiments, data processing, and model training (e.g., HPC clusters, GPU fleets, or cloud‑based systems) Enable researchers to go from idea to large‑scale experiment in minutes, not days, by building powerful abstractions for job submission, scheduling, and monitoring. Create tooling that improv

Staff Software Engineer - AI Research Infrastructure P-1215 At Databricks, we are obsessed with enabling data teams to solve the world’s toughest problems, from security threat detection to cancer drug development. We do this by building and running the world’s best data and AI platform so our customers can focus on the high-value challenges that are central to their own missions. The Databricks AI Research organization enables companies to develop AI models and agents using their own data, with technologies ranging from post-training open source LLMs to developing advanced multi-agent architectures. Databricks AI does so by producing novel science and putting it into production. Databricks AI is committed to the belief that a company’s AI models and agents are just as valuable as any other core IP, and that high-quality AI should be available to all. Job Description As a Staff Software Engineer, AI Research Infrastructure, you will be developing and running the research stack that powers Databricks AI Research. You will design and build services that schedule, orchestrate, and observe large‑scale training and inference experiment workloads across thousands of GPUs, improve our dev tooling and ensure that researchers can iterate quickly without sacrificing reliability, efficiency, or security. You’ll partner closely with research scientists, ML engineers, and platform teams to turn experimental workloads into robust, repeatable pipelines, and to push the limits of what our infrastructure can support. The Impact you will have As a Staff Software Engineer on the AI Research Infra Team at Databricks, you will: Design and implement infrastructure that supports large‑scale experiments, data processing, and model training (e.g., HPC clusters, GPU fleets, or cloud‑based systems) Enable researchers to go from idea to large‑scale experiment in minutes, not days, by building powerful abstractions for job submission, scheduling, and monitoring. Create tooling that improv

P-1428 At Databricks, we are passionate about enabling data and AI teams to solve the world's toughest problems — from making the next mode of transportation a reality to accelerating the development of medical breakthroughs. We do this by building and running the world's best data and AI infrastructure platform so our customers can use deep data insights to improve their business. Founded by engineers — and customer obsessed — we leap at every opportunity to solve technical challenges, from designing next-gen UI/UX for interfacing with data to scaling our services and infrastructure across millions of virtual machines. And we're only getting started. As part of the AI Platform team, you’ll build the substrate that powers everything from data apps, AI agents, model training, model serving, and vector search. You’ll be joining a high-agency, high-visibility team operating at the frontier of AI infrastructure — with deep ties to research, product, and real-world enterprise use cases. Databricks Mosaic AI is one of our fastest-growing businesses helping thousands of our customers democratize AI within their organizations. We’re building the infrastructure that powers the next generation of AI. We’re hiring across multiple teams in our AI Engineering org including: The impact you will have: Build infrastructure that powers our flagship offerings like MLflow, AI Gateway, Databricks Apps, Agent Framework, Agent Bricks, and Foundation Model APIs, to state a few. Improve reliability, latency, and efficiency of distributed AI workloads Collaborate with platform, infra, and ML teams to deliver seamless end-to-end experiences Shape how developers and data scientists build and interact with AI on Databricks What we look for: 5+ years of experience in backend or infrastructure engineering Strong programming skills in Scala, Go, or Python Experience with distributed systems, scalable APIs, or cloud-native infrastructure Familiarity with service

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? This role is for people who love building tools for their coworkers. The Internal Applications team creates tools that help us create better models. In this role you will collaborate with internal stakeholders, which include annotators, ML researchers, product managers and more. Join our team of builders who create tooling that will pave the way for the next generation of large language models! As a Full-Stack Software Engineer on the Internal Applications team, you will: Work with a small talented and enthusiastic team of software engineers Contribute to delightful experiences for our user-facing products, meticulously crafting code for browsers and servers Collaborate and grow with your engineering colleagues of all levels through direct pairing sessions, architectural designs, documentation and talks Identify and remove roadblocks to enable your team to increase its engineering velocity. Build resilient systems that are mission-critical Keep up with the cutting edge and adopt new technologies to improve performance and reliability You may be a good fit if: You have experience shipping products with a large numb

Reddit is a community of communities. It’s built on shared interests, passion, and trust, and is home to the most open and authentic conversations on the internet. Every day, Reddit users submit, vote, and comment on the topics they care most about. With 100,000+ active communities and approximately 130 million daily active unique visitors, Reddit is one of the internet’s largest sources of information. For more information, visit www.redditinc.com . Reddit has a flexible first workforce! We’re looking for a Staff-level Machine Learning Engineer to join Reddit’s Ads Ranking Org. This role is part of a broader effort to enhance Reddit’s ML-powered ad ranking systems through improved model architectures, contextual embeddings, and conversion-optimized modeling. The ideal candidate will bring strong end-to-end modeling expertise, a deep understanding of modern deep learning techniques, and a passion for building systems that drive measurable impact in ads marketplaces. While ads experience is a plus, strong modelers from recommendation systems, search relevance, or other performance-driven domains are encouraged to apply. This position will sit on the App Ads Modeling team and report directly into the Ranking Org. Responsibilities: Design and train advanced ML models (e.g., DNNs, transformers) to power Reddit Ads Ranking Develop and optimize features including embeddings, contextual signals, and cross-session behavior Work closely with product, infra, and data teams to drive end-to-end model deployment and performance analysis Mentor other MLEs and contribute to modeling best practices across the org Shape the long-term modeling vision across one or more domains (conversion, app ads, shopping, brand, etc.) Required Qualifications: 7+ years of industry experience, including several years in applied ML roles Strong experience with Deep Learning architectures and ML frameworks (TensorFlow, PyTorch) Strong background in recommendation systems, ads ranking, or similar

About Decagon Decagon is the leading conversational AI platform empowering every brand to deliver concierge customer experiences. Our technology enables industry-defining enterprises like Avis Budget Group, Block’s Cash App and Square, Chime, Oura Health, and Hunter Douglas to deploy AI agents that power personalized, deeply satisfying interactions across voice, chat, email, SMS, and every other channel. We’re building a future where customer experiences are being redefined from support tickets and hold music to faster resolutions, richer conversations, and deeper relationships. We’re proud to be backed by world-class investors who share that vision, including a16z, Accel, Bain Capital Ventures, Coatue, and Index Ventures, along with many others. We’re an in-office company, driven by a shared commitment to excellence and velocity. Our values — Just Get It Done, Invent What Customers Want, Winner’s Mindset, and The Polymath Principle — shape how we work and grow as a team. About the Team The Infrastructure team builds and operates the foundations that power Decagon: networking, data, ML serving, developer platform, and real‑time voice. We partner closely with product, data, and ML to deliver high‑scale, low‑latency systems with clear SLOs and great developer ergonomics. Read more about the infra team's work here: https://decagon.ai/blog/what-an-air-gapped-ai-deployment-actually-requires About the Role Decagon builds agentic AI that resolves customer support conversations end to end, for companies ranging from fast-growing startups to some of the largest financial institutions in the world. Keeping that system fast, reliable, and secure (across our multi-tenant cloud and inside customers' own locked-down environments) is an infrastructure problem, and that's the problem this role owns. You'll build the platforms and abstractions our product teams ship on, and you'll architect and operate the deployments that run our agents inside enterprise customer clouds, where secu

About the Team The Experimentation Platform Team develops a state-of-the-art platform in industry that enables Product Engineers, Data Scientists, ML Engineers and non-technical audiences to come up with hypotheses; design, configure and analyze experiments; and conduct exploratory and causal analysis. At DoorDash, where we run thousands of experiments per year, our mission is to equip all decision makers with rigorous, data-driven insights by democratizing experimentation with quality and velocity. The team consists of a mix of experienced veterans of backend, web, statistical and data infra engineers and work closely with the data science community. Some of the interesting work done in the team was published in articles such as: Meet Dash-AB-The Statistics Engine of Experimentation at DoorDash Supporting Rapid Product Iteration with an Experimentation Analysis Platform The 4 Principles DoorDash Used to Increase Its Logistics Experiment Capacity by 1000% Improving Online Experiment Capacity by 4X with Parallelization and Increased Sensitivity About the Role If you want to solve the toughest engineering challenges, build cutting-edge experimentation products under rigorous operational constraints (high volume, correctness guarantees) and work with some of the smartest people in the industry, DoorDash’s Experimentation Platform is the right place for you. As a full stack engineer, you’ll build across our backend services, SDKs, and the Decision Portal UI used by thousands of internal users. Come join us and be part of the mission. You will report into the engineering manager on our Decision Experience team as part of the Decision Systems team. We expect this role to be hybrid (San Francisco, Sunnyvale, Seattle, New York) with some time in-office and some time remote. What You'll Do Work on dramatically enhancing and simplifying the Experimentation platform which is used by almost every engineer in the company. Have the opportunity to build a new Experimentation platf

About Anyscale: At Anyscale , we're on a mission to democratize distributed computing and make it accessible to software developers of all skill levels. We’re commercializing Ray , a popular open-source project that's creating an ecosystem of libraries for scalable machine learning. Companies like OpenAI , Uber , Spotify , Instacart , Cruise , and many more, have Ray in their tech stacks to accelerate the progress of AI applications out into the real world. With Anyscale, we’re building the best place to run Ray, so that any developer or data scientist can scale an ML application from their laptop to the cluster without needing to be a distributed systems expert. Proud to be backed by Andreessen Horowitz, NEA, and Addition with $250+ million raised to date. About the role: Anyscale is looking for a Site Reliability Engineer to join the Infrastructure team. Anyscale aims to provide the next generation of tools and infrastructure to make developing and running distributed AI applications in the cloud as easy as on your laptop. As part of the Infra team, we build the scalable, secure, and robust backbone that enables this vision. Our team is responsible for both the control plane, which orchestrates cluster management, scheduling, and user access, and the data plane, which ensures high-performance execution of distributed workloads. We are seeking a talented engineers with a strong background in control plane and data plane development, along with expertise in Kubernetes, container orchestration, and cloud-native infrastructure. You will play a crucial role in designing, implementing, and optimizing the critical infrastructure that powers Anyscale’s cloud platform. You will have the opportunity to work on open-source Ray, contribute to our infinite laptop proprietary product, and develop seamless integration between the two, while also delivering high-impact features for our customers. A snapshot of projects you may work on Design, build, and scale services that orches

About Anyscale: At Anyscale , we're on a mission to democratize distributed computing and make it accessible to software developers of all skill levels. We’re commercializing Ray , a popular open-source project that's creating an ecosystem of libraries for scalable machine learning. Companies like OpenAI , Uber , Spotify , Instacart , Cruise , and many more, have Ray in their tech stacks to accelerate the progress of AI applications out into the real world. With Anyscale, we’re building the best place to run Ray, so that any developer or data scientist can scale an ML application from their laptop to the cluster without needing to be a distributed systems expert. Proud to be backed by Andreessen Horowitz, NEA, and Addition with $250+ million raised to date. About the role: Anyscale is looking for a Software Engineer to join the Infrastructure team. Anyscale aims to provide the next generation of tools and infrastructure to make developing and running distributed AI applications in the cloud as easy as on your laptop. As part of the Infra team, we build the scalable, secure, and robust backbone that enables this vision. Our team is responsible for both the control plane, which orchestrates cluster management, scheduling, and user access, and the data plane, which ensures high-performance execution of distributed workloads. We are seeking a talented Software Engineer with a strong background in control plane and data plane development, along with expertise in Kubernetes, container orchestration, and cloud-native infrastructure. You will play a crucial role in designing, implementing, and optimizing the critical infrastructure that powers Anyscale’s cloud platform. You will have the opportunity to work on open-source Ray, contribute to our infinite laptop proprietary product, and develop seamless integration between the two, while also delivering high-impact features for our customers. A snapshot of projects you may work on Design, build, and scale services that orches

Every day, tens of millions of people come to Roblox to explore, create, play, learn, and connect with friends in 3D immersive digital experiences– all created by our global community of developers and creators. At Roblox, we’re building the tools and platform that empower our community to bring any experience that they can imagine to life. Our vision is to reimagine the way people come together, from anywhere in the world, and on any device. We’re on a mission to connect a billion people with optimism and civility, and looking for amazing talent to help us get there. A career at Roblox means you’ll be working to shape the future of human interaction, solving unique technical challenges at scale, and helping to create safer, more civil shared experiences for everyone. Roblox's data infrastructure processes petabytes of data daily, powering analytics, ML, and product decisions for a platform serving 200M+ daily active users. As a Principal Software Engineer in our Data Infra org, you will be the primary technical leader driving the strategic vision, long-term architecture, and massive scalability of our distributed data platforms that power Roblox. You will own and drive the next-generation architecture of our core platforms, which span Kafka, Flink, Spark, Trino, Druid, Airflow and Data Catalog. This role operates under high ambiguity, demanding unparalleled ownership to redefine the limits of infrastructure handling exabyte-scale workloads, and providing a unique opportunity to lead the future evolution of our global data ecosystem. You Will: Define Multi-Year Technical Strategy: Own and drive the end-to-end architectural vision for Roblox's core data platforms spanning Kafka, Flink, Spark, Trino, Druid, Airflow, and Data Catalog systems. Turn multi-year company strategies into concrete, production-grade infrastructure blueprints. Lead Cross-Functional Alignment: Partner closely with executive leadership, platform governance, data science, and product e

About the Team At DoorDash, we’re reimagining how people connect with the things they need — whether it’s a meal, a grocery run, and anything in between. Our audiences — Consumers, Dashers, and Merchants — are at the heart of everything the Design org builds. Our content design team plays a big part in this, with each content designer shaping the experience and bringing our vision to life through clear, thoughtful language that makes our three-sided marketplace easier, faster, and more human. About the Role We’re looking for a full stack UX Design Engineer to be the technical backbone of our global Content Design org and leader who builds the platforms, tooling, and ML systems that make world-class, localized product content effortless across DoorDash, Wolt, and Deliveroo. You’ll work at the intersection of design and engineering to build content tooling and embed LLM-driven workflows into product experiences. You’ll accelerate content design AI tooling so that CD’s can focus on the highest leverage strategic work, while your tools enable product designers and cross functional partners to ship high-quality content faster across DoorDash, Wolt, and Deliveroo. You're excited about this opportunity because... Design and build internal content tools that: Help PMs and designers generate, manage, localize, and deploy product content at scale. Plug custom GPTs and other LLMs into everyday product workflows for content iteration and improvements. Own full-stack development of these tools: Build and maintain frontend experiences using React and modern JavaScript/TypeScript. Design and implement backend services and APIs (e.g., Kotlin/Java) to support content tooling, experimentation, and automation. Integrate tooling with experimentation and infra: Embed content tools into A/B testing platforms so teams can test and ship variants with minimal engineering dependency. Build pipelines from variant generation → experiment → automated deployment of winners. Operationalize

P-1591 At Databricks, we are passionate about enabling data and AI teams to solve the world's toughest problems — from making the next mode of transportation a reality to accelerating the development of medical breakthroughs. We do this by building and running the world's best data and AI infrastructure platform so our customers can use deep data insights to improve their business. Founded by engineers — and customer obsessed — we leap at every opportunity to solve technical challenges, from designing next-gen UI/UX for interfacing with data to scaling our services and infrastructure across millions of virtual machines. And we're only getting started. As part of the AI Platform team, you’ll build the substrate that powers everything from data apps, AI agents, model training, model serving, and vector search. You’ll be joining a high-agency, high-visibility team operating at the frontier of AI infrastructure — with deep ties to research, product, and real-world enterprise use cases. Databricks Mosaic AI is one of our fastest-growing businesses helping thousands of our customers democratize AI within their organizations. We’re building the infrastructure that powers the next generation of AI. We’re hiring across multiple teams in our AI Engineering org including: The impact you will have: Build infrastructure that powers our flagship offerings like MLflow, AI Gateway, Databricks Apps, Agent Framework, Agent Bricks, and Foundation Model APIs, to state a few. Improve reliability, latency, and efficiency of distributed AI workloads Collaborate with platform, infra, and ML teams to deliver seamless end-to-end experiences Shape how developers and data scientists build and interact with AI on Databricks What we look for: 5+ years of experience in backend or infrastructure engineering Strong programming skills in Scala, Go, or Python Experience with distributed systems, scalable APIs, or cloud-native infrastructure Familiarity with service-oriented a

Figma is growing our team of passionate creatives and builders on a mission to make design accessible to all. Figma’s platform helps teams bring ideas to life—whether you're brainstorming, creating a prototype, translating designs into code, or iterating with AI. From idea to product, Figma empowers teams to streamline workflows, move faster, and work together in real time from anywhere in the world. If you're excited to shape the future of design and collaboration, join us! We're looking for a research-minded Data Scientist to join the Core Data team. This team is a group of analytics professionals and Engineers building the foundational platforms for data science at Figma. We build the experimentation, analytics, and AI tooling that every product team relies on to make confident, data-driven decisions, partnering closely with Data Infra, ML, and Applied Science to evolve our platforms and embed AI into the daily workflows of data scientists across the company. This role is for someone who thrives at the intersection of rigorous research and real-world impact. You'll bring PhD-level depth to problems that matter. This includes advancing our experimentation platform and developing machine learning-based analytical systems. You will also help craft how we measure AI-powered features through causal inference and statistical modeling. This is a full time role that can be held from one of our US hubs or remotely in the United States. What you'll do at Figma: Partner across teams to define and track important metrics, develop experiments, and uncover insights that inform strategic decisions Accelerate Figma's experimentation platform and methodology, including A/B testing frameworks and causal inference techniques Construct models and analytical frameworks based on machine learning to support product, platform, and business initiatives Create tools, datasets, and systems that enable others to work with data more efficiently

Every day, tens of millions of people come to Roblox to explore, create, play, learn, and connect with friends in 3D immersive digital experiences– all created by our global community of developers and creators. At Roblox, we’re building the tools and platform that empower our community to bring any experience that they can imagine to life. Our vision is to reimagine the way people come together, from anywhere in the world, and on any device. We’re on a mission to connect a billion people with optimism and civility, and looking for amazing talent to help us get there. A career at Roblox means you’ll be working to shape the future of human interaction, solving unique technical challenges at scale, and helping to create safer, more civil shared experiences for everyone. The Data Center Network Engineering (DNE) is a pod within the Physical Networking team and owns the design, deployment, and operation of Roblox's large-scale data center network. This includes server-layer connectivity, intra and inter-DC connectivity. The team is building High-Performance Networking (HPN) fabrics, deploying 400G/800G networking, and scaling GPU clusters to support AI training and inference. You will: Drive the design, deployment, and operations of Roblox's large-scale data center network, including the networking for GPU clusters that support training and inference workloads. Design, qualify, and validate new technologies and hardware for scalable solutions. Coordinate data center builds and projects with multiple teams, including weekly cross-team calls. Work with Network Reliability Systems, Data Center Ops, Capacity Planning, Supply Chain, and Compute/ML teams to address network challenges early. Contribute to automation and telemetry that improve efficiency, reliability, and visibility. You have: 6+ years of expertise designing and deploying large-scale data center fabrics (CLOS). Experience with GPU clusters or AI/ML networking. Proficiency deploying and managin

Every day, tens of millions of people come to Roblox to explore, create, play, learn, and connect with friends in 3D immersive digital experiences– all created by our global community of developers and creators. At Roblox, we’re building the tools and platform that empower our community to bring any experience that they can imagine to life. Our vision is to reimagine the way people come together, from anywhere in the world, and on any device. We’re on a mission to connect a billion people with optimism and civility, and looking for amazing talent to help us get there. A career at Roblox means you’ll be working to shape the future of human interaction, solving unique technical challenges at scale, and helping to create safer, more civil shared experiences for everyone. The Data Center Network Engineering (DNE) is a pod within the Physical Networking team and owns the design, deployment, and operation of Roblox's large-scale data center network. This includes server-layer connectivity, intra and inter-DC connectivity. The team is building High-Performance Networking (HPN) fabrics, deploying 400G/800G networking, and scaling GPU clusters to support AI training and inference. You will: Drive the design, deployment, and operations of Roblox's large-scale data center network, including the networking for GPU clusters that support training and inference workloads. Design, qualify, and validate new technologies and hardware for scalable solutions. Coordinate data center builds and projects with multiple teams, including weekly cross-team calls. Work with Network Reliability Systems, Data Center Ops, Capacity Planning, Supply Chain, and Compute/ML teams to address network challenges early. Contribute to automation and telemetry that improve efficiency, reliability, and visibility. You have: 6+ years of expertise designing and deploying large-scale data center fabrics (CLOS). Experience with GPU clusters or AI/ML networking. Proficiency deploying and managin