Software Engineer - Machine Learning Infrastructure

About ZaimlerAt Zaimler, we believe that AI agents need a profound understanding of data to operate effectively. In today’s fragmented enterprise data landscape, where information is scattered across various systems without a unified context, traditional AI solutions often fall short. We are pioneering the transition from basic copilots to fully autonomous agents, creating a revolutionary infrastructure layer that makes this transition possible.Our platform serves as the foundational context infrastructure for the agentic era. We automatically uncover domain knowledge, establish meaningful relationships, and endow AI agents with the semantic comprehension necessary for precise operations at scale. Imagine knowledge graphs that not only facilitate real-time inference but are also designed for systems that require reasoning capabilities beyond mere retrieval.Founded by industry veterans Biswajit Das, former VP of Engineering at Truera and Chief Architect at Visa, and Sofus Macskassy, ex-Director of Engineering at LinkedIn and creator of one of the largest production knowledge graphs, Zaimler is a close-knit, senior team currently in the seed stage. We are collaborating with major enterprises across sectors such as insurance, travel, and technology. If you are passionate about shaping the infrastructure that will power the next decade of AI innovation, we would love to connect with you.About the RoleAs part of our Machine Learning team, you will focus on transforming raw enterprise data into structured, contextualized knowledge graphs and embeddings. Your responsibilities will include developing innovative and scalable algorithms for machine learning and data engineering aimed at enhancing system efficiency. You will also explore new methodologies to compress large models into more efficient versions, improve retrieval and reasoning performance through feedback mechanisms, and prototype techniques that enable large language models (LLMs) to extract and utilize real-world knowledge efficiently.We seek a candidate who thrives in a fast-paced environment, values meticulous work, and is eager to learn from some of the top engineers and researchers in the field.

Roblox is a dynamic platform where millions of users venture daily to explore, create, engage, and connect with friends in immersive 3D digital environments crafted by our vibrant community of developers and creators.At Roblox, we are dedicated to developing tools and platforms that empower our community to transform their imaginative ideas into reality. Our vision is to redefine global connectivity, enabling people to come together from anywhere in the world, accessible on any device. We are on a mission to unite a billion people with positivity and respect, and we are seeking exceptional talent to help us achieve this goal.A career at Roblox means you will play a pivotal role in shaping the future of human interaction, tackling unique technical challenges on a grand scale, and contributing to the creation of safer, more respectful shared experiences for all.The Foundation AI Group aims to position Roblox as the benchmark for 3D foundational models (3DFMs), making it easier for anyone to craft high-quality, immersive 3D experiences using AI. The AI Platform team is integral to this vision, managing hundreds of machine learning use cases and handling billions of inferences daily across areas like Discovery, Safety, and Engine. We are searching for exceptional PhD graduates to spearhead innovation in three vital areas: AI Platform, Distributed Inference Systems, and Generative AI Information Retrieval.Your RoleAs a Senior Machine Learning Engineer on the AI Platform team, you will be a vital contributor to the development of innovative systems that drive AI at Roblox. You will concentrate on one of three significant tracks:Track 1: AI Platform ProjectsLead the development of next-generation AI tools that improve the efficiency, cost, and usability of ML@Roblox.Build and sustain essential platform components: Serving Layer, Model Registry, Pipeline Orchestrator, and Training/Inference control planes.Craft outstanding developer experiences (templates, tools, visualizations) to reduce time-to-production and ensure foundational AI systems are scalable and reliable.Track 2: Distributed Inference & Systems OptimizationDesign and implement scalable distributed inference systems for efficiently serving large language models (LLMs) and Large Recommender Models at unprecedented scale.Perform in-depth, low-level performance analysis to optimize systems.

At Roblox, we welcome millions of users daily to engage in immersive 3D experiences crafted by our vibrant community of developers and creators.Our mission is to empower this community by providing innovative tools and a platform that turn their imaginative ideas into reality. We aim to transform the way people connect globally, across all devices, fostering a spirit of optimism and civility as we work towards our goal of connecting one billion users.Joining Roblox means you will play a key role in shaping the future of human interaction, solving significant technical challenges, and enhancing shared experiences for all.As our user base grows at an unprecedented rate, we are on the lookout for proficient machine learning engineers who excel in tackling intricate challenges and developing scalable, cutting-edge solutions. In this influential role, you will create deep learning models focused on user insights, content analysis, and recommendation systems, thereby enhancing user engagement, retention, and growth in our creator ecosystem.Your contributions will be vital in refining the core systems that facilitate user discovery and interaction across Roblox. By enhancing exploration and engagement, you will help users find relevant and exciting content. Looking to the future, our team is dedicated to innovating personalized recommendation algorithms and advancing CV and NLP techniques to deepen our understanding of Roblox experiences.Join us and be a driving force in connecting a billion users with the optimal content and interactions at the perfect moment, addressing diverse challenges on an unprecedented scale.

About the RoleJoin our dynamic team, affectionately known as MBMB (More Big More Better), where you will play a crucial role in optimizing our training and on-robot inference stacks. We are seeking bold innovations that drive substantial improvements rather than incremental changes.Your Responsibilities Will Include:Maximizing GPU performance through innovative strategiesDeploying machine learning, hardware, and software modifications that yield significant advancementsEnhancing both inference and training stacks for optimal performanceIdeal Candidates Will:Possess proficiency in the latest machine learning techniques, particularly for training and inference optimizations within transformer and diffusion-based architecturesHave a relentless pursuit of ML optimizations across various domains, including CUDA kernels, ML architecture, frontend and backend network bottlenecks, CPU inefficiencies, NVLink, and communication protocols, as well as optimizations in libraries such as Torch, NumPy, and Python.

Every day, tens of millions of users engage with Roblox to explore, create, play, learn, and connect with friends through immersive 3D digital experiences crafted by our global community of developers and creators. At Roblox, we are developing the tools and platform that empower our community to bring any experience they can envision to life. Our mission is to transform the way individuals connect, from anywhere in the world, and on any device. We aspire to unite a billion people with positivity and respect, and we are seeking exceptional talent to help us achieve this goal. A career at Roblox means you will contribute to the future of human interaction, tackle unique technical challenges at scale, and help foster safer, more respectful shared experiences for all.As a Senior Machine Learning Engineer on the Economy ML team, you will develop models that enhance ranking, personalization, pricing, and optimization across Roblox’s economic ecosystems, including the Marketplace, Developer Monetization, and Payments. Your contributions will significantly influence how users find virtual items, how creators earn revenue, and how the platform evolves sustainably.The Economy ML team is constructing the machine learning foundation for Roblox’s virtual economy—from smart pricing and personalized storefronts to dynamic layout optimization and understanding avatar content. We closely collaborate with Marketplace, Payments, and Developer Monetization teams to create ML systems that drive long-term value across engagement, retention, and creator earnings.Your Responsibilities:Translate vague product challenges into actionable designs and deliver end-to-end ML solutions for ranking, recommendation, pricing, and content comprehension across economic platforms.Construct and refine deep learning retrieval and ranking models that personalize item and widget ordering across the Marketplace and other shopping interfaces.Create models and policies that optimize pricing, conversion, and revenue while maintaining ecosystem health, enhancing user experience, and ensuring creator success.Oversee the complete ML lifecycle for your projects: data exploration, feature engineering, model design, training, evaluation, and production deployment on Roblox’s ML platform.Design and execute robust testing frameworks to validate model performance and business impact.

Join the vibrant world of Roblox, where millions of users engage in limitless exploration, creativity, and social interaction through immersive 3D digital experiences, all crafted by our dynamic global community of developers and creators.At Roblox, we are dedicated to constructing the tools and platform that enable our community to transform their imaginative ideas into reality. Our mission is to revolutionize how individuals connect, regardless of their location or device. We aim to unite a billion people with hope and respect, and we are seeking exceptional talent to join us on this journey.A career at Roblox means playing a pivotal role in shaping the future of human interaction, tackling unique technical challenges at scale, and fostering safer, more respectful shared experiences for all.Who We Are:Help shape the future of Roblox’s virtual economy.The Economy ML team is at the forefront of developing the machine learning infrastructure that drives Roblox’s Marketplace, Developer Monetization, and Payments systems. Our initiatives include intelligent pricing strategies, personalized storefronts, dynamic layout optimization, and enhanced avatar understanding, all designed to transform user engagement, monetization, and creator success on a grand scale.As a Principal Software Engineer (Data Systems), you will be responsible for architecting, building, and deploying robust, high-scale real-time and batch data systems aimed at enhancing personalization, search, and recommendation functionalities across Marketplace, Developer Monetization, and Payments products. You will contribute to crucial data projects, from designing event taxonomies and logging interfaces to real-time feature serving across various search and recommendation platforms.Your ResponsibilitiesLead data engineering efforts for the Economy ML team, establishing standards for batch versus streaming feature pipelines, table design, observability, and documentation across the Economy division.Actively engage as a hands-on contributor to our data systems, driving content recommendation, search, and personalization across Economy product surfaces.Collaborate with Product, Data Science, and ML engineering teams to prioritize data initiatives that align with key business objectives and mitigate technical debt in existing datasets and pipelines.Guide and mentor fellow DE/ML engineers on best practices for schema design, client logging, and more.

About zaimlerAt zaimler, we recognize that AI agents struggle to make decisions based on fragmented data. Today’s enterprise data is scattered across numerous systems, lacking shared context and structure, which contributes to the challenges faced in enterprise AI. As we transition from copilots to fully autonomous agents, we're establishing a revolutionary infrastructure layer to support this evolution.zaimler serves as the foundational context infrastructure for the agentic era. Our platform autonomously uncovers domain knowledge, maps intricate relationships, and equips AI agents with the semantic understanding necessary for precise operations at scale. Envision knowledge graphs that facilitate real-time inference, designed specifically for systems that require reasoning capabilities, rather than mere data retrieval.Founded by industry veterans Biswajit Das and Sofus Macskassy, zaimler is a small yet highly skilled team at the seed stage, collaborating with major enterprises across sectors such as insurance, travel, and technology. If you aspire to develop the infrastructure that will underpin the next decade of AI advancements, we invite you to join our journey.About the RoleWe are in search of a Senior Machine Learning Engineer to enhance our dynamic team, ideally based in the Bay Area or open to relocation. The perfect candidate will possess strong expertise in one or more of the following areas: Knowledge Extraction, Natural Language Understanding, Unsupervised Learning, Information Retrieval, and Fine-tuning Large Language Models (LLMs). In this pivotal role, you will be instrumental in developing and training the models, pipelines, and methodologies that drive our semantic graph systems. We seek an individual with a robust background in machine learning, natural language processing, LLMs, and semantic technologies, along with a proven history of managing complex, large-scale machine learning projects.

At Roblox, millions of individuals engage daily to explore, create, play, learn, and connect with friends in immersive 3D digital experiences crafted by our global community of developers and creators.We are dedicated to building tools and platforms that empower our community to realize their imaginative experiences. Our vision is to transform how people connect from anywhere in the world and on any device. Our mission is to unite one billion people with optimism and civility, and we are seeking exceptional talent to help us achieve this goal.Joining Roblox means you will play a pivotal role in shaping the future of human interaction, tackling unique technical challenges at scale, and contributing to the creation of safer, more civil shared experiences for all.The Rights Manager and Content Suitability teams, part of our Safety Experience, are in search of a Senior Machine Learning Engineer.The Safety Experience organization develops tools and systems that empower Roblox users and creators to control their experiences while enabling moderators to uphold our community standards. Our focus includes education, intervention, visibility, and action.Our initiatives include:Monitoring and influencing user behavior for enhanced safety.Building scalable and efficient moderation platforms.Providing transparency and educational resources for parents and developers.Empowering users to manage their own safety.Allowing IP owners to manage their creations on Roblox.Delivering quick and accurate support through our Customer Care Chatbot.Your responsibilities will include:Implementing machine learning solutions for safety-focused systems.Promoting a culture of technical excellence and inclusivity.Breaking down long-term product requirements into actionable phases for continuous improvement.Designing and building large-scale machine learning models with billions of parameters, ensuring they are production-ready.Facilitating complex technical decisions with empathy and foresight.

About Us:At Fireworks AI, we are pioneering the future of generative AI infrastructure. Our platform is recognized for delivering the highest-quality models with the fastest and most scalable inference capabilities in the industry. Proudly benchmarked as a leader in LLM inference speed, we are at the forefront of innovation with projects like function calling and multimodal models. As a Series C company valued at $4 billion, we are supported by prestigious investors including Benchmark, Sequoia, Lightspeed, Index, and Evantic. Our team comprises ambitious builders, many of whom hail from Meta PyTorch and Google Vertex AI.The Role:As an Applied Machine Learning Engineer, you will play a crucial role in bridging the gap between advanced AI research and practical applications. Your responsibilities will include developing, fine-tuning, and operationalizing machine learning models that deliver significant business value and enhance user experiences. This hands-on engineering position demands a blend of deep technical expertise and a strong customer-centric approach to create scalable AI solutions.Key Responsibilities:Customer Success: Collaborate with the Go-To-Market (GTM) team, including Account Executives and Solutions Architects, to ensure seamless integration and successful deployment of machine learning solutions.Demo / Proof of Concept (PoC): Develop and present engaging PoCs that showcase the capabilities of our AI technology.Application Build: Design, develop, and deploy end-to-end AI-powered applications customized to meet customer needs.Platform Features / Bug Fixes: Contribute to the internal machine learning platform by adding features and resolving issues.New Model Enablements: Integrate and enable new machine learning models within the existing platform or client environments.Performance Optimizations: Enhance the performance, efficiency, and scalability of deployed models and applications.Partnership Enablement: Collaborate closely with partners to facilitate joint AI solutions and ensure effective collaboration.

Roblox is a vibrant platform where millions of users engage daily to explore, create, play, learn, and connect through immersive 3D experiences, all crafted by our diverse community of developers and creators.At Roblox, we are dedicated to building innovative tools and platforms that empower our community to turn their imaginative experiences into reality. Our vision is to revolutionize how people connect from anywhere in the world, across any device. We are on a purposeful mission to unite a billion individuals with positivity and respect, and we are seeking exceptional talent to help us achieve this goal.Joining Roblox means contributing to the evolution of human interaction, tackling unique technical challenges at scale, and fostering safer, more respectful shared experiences for all.In the role of Distinguished Machine Learning Engineer/Technical Director within the Safety organization at Roblox, you will spearhead the technical vision and execution of machine learning projects aimed at ensuring user safety and promoting civility. Our top-tier safety features are what make Roblox a secure and inclusive space for creative expression and shared experiences. The Safety team plays a critical role in maintaining Roblox as one of the safest platforms on the internet, protecting users as our platform grows to accommodate diverse age groups and locations.You Will:Take ownership of the technical direction and implementation of machine learning solutions for safety systems.Lead and mentor fellow engineers, fostering an environment of technical excellence and inclusivity.Translate long-term product requirements into manageable milestones, ensuring ongoing enhancements.Design and develop large-scale machine learning models with billions of parameters, ensuring they are production-ready.Navigate complex technical decisions across teams, showcasing empathy and fostering mutual understanding.Work collaboratively with cross-functional teams to outline and prioritize the machine learning roadmap.

Join Our Team as a Senior ML Research Engineer, Foundation ModelsAbout UsBecome part of a cutting-edge team at Genesis Molecular AI, where we are dedicated to pioneering advancements at the intersection of artificial intelligence and biochemistry. Our accomplished group comprises distinguished deep learning researchers, software engineers, and innovators in drug discovery.Our mission is transformative: to develop innovative AI foundation models that provide breakthrough therapies for patients with critical health challenges. We conduct essential research that amalgamates machine learning, physics, and computational chemistry, continuously pushing the boundaries of these disciplines.At Genesis AI, we are constructing the core infrastructure for this revolutionary work. We design and implement large-scale generative models, leveraging comprehensive molecular data supported by robust computational resources and simulation pipelines.About the RoleWe are seeking a highly proficient ML Research Engineer who excels in bridging fundamental research with production-level engineering.As an integral member of the Genesis AI team, you will be instrumental in conceptualizing, scaling, and deploying our next generation of foundation models for molecular science.Collaboration is key, as you will work closely with ML researchers, computational chemists, and drug discovery scientists to translate pioneering model concepts into actionable systems that drive real drug discovery initiatives.Key ResponsibilitiesEnhancing and scaling model pretraining pipelinesAdvancing reinforcement learning and post-training systemsOptimizing the performance of extensive molecular modelsImplementing structure prediction models like Pearl into production settings utilized by chemists and drug programsThis position requires an individual capable of integrating ML and computational chemistry, facilitating swift transitions from research insights to practical applications.

About the TeamBecome a part of a groundbreaking team at the intersection of artificial intelligence and biochemistry.At Genesis Molecular AI, we have assembled a close-knit team of distinguished deep learning researchers, software engineers, and pioneers in drug discovery. Our ambitious mission is to develop the next generation of AI foundation models that will enable innovative therapies for patients suffering from serious illnesses.We go beyond the application of machine learning in biology; we conduct pioneering research at the confluence of machine learning, physics, and computational chemistry, continuously pushing the limits of each discipline. You will collaborate with top-tier multidisciplinary researchers to design and implement scalable generative foundation models, backed by extensive computational resources and large-scale simulations.About the RoleThis exceptional opportunity is tailored for a scientist passionate about leveraging advanced AI to tackle real-world challenges in drug discovery. You will play a pivotal role in connecting our long-term research initiatives with our active drug discovery programs. Your objective will be to construct, assess, monitor, and refine our cutting-edge models, integrating them directly into ongoing drug programs and leading the efforts in model validation, deployment, and analysis to inform the discovery of new medications.As both a translator and strategist, you will ensure that our research targets the most pressing challenges and that our drug hunters can harness the full capabilities of our leading AI platform. This role necessitates an in-depth understanding of cheminformatics, computational chemistry, and experimental techniques, robust data science skills, and an aptitude for articulating complex concepts to a diverse, multidisciplinary team.Positions are available at multiple seniority levels: Senior, Staff, and Principal.

About WisdomAIAt WisdomAI, we are committed to democratizing access to data and insights for everyone. We leverage the transformative potential of Generative AI to revolutionize how organizations utilize data for informed decision-making. Join us now, as we partner with multiple Fortune 100 companies and embark on an exciting journey to shape the future of data accessibility and understanding.Position OverviewWe are on the lookout for talented Machine Learning and Natural Language Processing engineers to join our dynamic team. You will tackle core challenges in large language model (LLM) code generation, document comprehension, context synthesis, and the development of LLM-based agents. Our team is comprised of seasoned professionals who hail from industry leaders such as Glean, Apple, Rubrik, Google, and Meta.Key ResponsibilitiesDevelop workflows and agentic pipelines to handle complex analytical tasks.Design and implement an evaluation framework for our query processing engine.Collaborate closely with customers and founders to establish the product and engineering roadmap.Influence the engineering culture and technology stack at WisdomAI.

About the RoleAt Generalist, we are at the forefront of training expansive robot foundation models, leveraging cutting-edge GPU hardware, primarily from Nvidia, to execute distributed training tasks and experimental research. Our operations demand exceptional storage solutions and optimized data loading processes, necessitating the full utilization of cloud infrastructure alongside custom-built solutions.In this role, you will take charge of our inference infrastructure. Our robotic systems rely on a dedicated fleet of on-premises GPUs designed for demanding real-time computations and latency-sensitive applications within resource-constrained environments.Your Responsibilities:Manage and optimize our GPU compute fleets.Facilitate user-friendly access to GPUs for researchers, ensuring optimal utilization.Enhance ML data loading, transport, and storage systems in extensively utilized distributed environments.Oversee the orchestration of our robot inference fleets.You May Excel in This Position If You:Have experience managing large GPU fleets for large-scale, distributed training or inference.Possess significant expertise in using Slurm or Kubernetes for ML workload orchestration.Have developed high-scale ML data loaders and preparation systems.Understand the intricacies of ML hardware, storage, and networking systems.Are familiar with the Nvidia GPU ecosystem.

Join Roblox as a Senior Machine Learning Engineering Manager and lead a dynamic team focused on developing cutting-edge machine learning solutions. In this role, you will be responsible for overseeing the design, implementation, and optimization of algorithms that enhance user experience and gameplay. Your leadership will guide the team in leveraging data to drive strategic decisions, ensuring our platform remains at the forefront of technology.

Join Our Innovative TeamAt Genesis Therapeutics, we are a dynamic and passionate group of drug discovery experts, deep learning researchers, and software engineers dedicated to revolutionizing biochemistry through AI. Our mission is clear: to uncover and develop transformative therapies for patients with severe medical conditions.Our AI team is at the forefront of creating foundational models for small molecule drug discovery. We conduct cutting-edge research that bridges machine learning, physics, and computational chemistry, while building resilient software systems capable of executing large-scale simulations and training advanced generative and predictive AI models utilizing our powerful cluster of thousands of GPUs and tens of thousands of CPUs.Your RoleWe are on the lookout for skilled ML infrastructure engineers to propel our machine learning research initiatives, particularly in generative modeling of molecular systems, which is vital to our overarching goals.In this position, you will spearhead the rapid advancement of our AI platform and infrastructure, enhancing performance, efficiency, and scalability to unprecedented levels. You will construct expansive distributed training and inference pipelines, essential MLOps tools and frameworks, and fine-tune GPU operations to accelerate ML model performance.Genesis fosters a collaborative and interdisciplinary environment, allowing you to work closely with our talented engineers, researchers, and scientists.Your ResponsibilitiesDrive engineering initiatives aimed at the continuous enhancement of our AI platform, focusing on the rapid development of scalable and robust distributed infrastructures for ML training, inference, and evaluation.Facilitate model training and deployment across various clusters and cloud environments, optimizing for throughput and cost-effectiveness.Maximize the efficiency of ML models and other workloads in terms of latency, throughput, and memory usage, particularly through GPU performance engineering, pushing the boundaries of current hardware capabilities.Contribute to the long-term strategic vision for Genesis’ infrastructure platform.Your ProfileA strong engineering background with a focus on machine learning infrastructure.

Join Roblox as a Principal Machine Learning Engineer focused on enhancing reliability across our systems. In this pivotal role, you will leverage cutting-edge machine learning techniques to ensure the seamless performance and reliability of our platform. Collaborate with cross-functional teams to design and implement robust ML algorithms that enhance user experience and system functionality.

ML Research Engineer, Foundation ModelsAbout Our TeamBecome part of an elite team that is pioneering advancements in AI and biochemistry. At Genesis Molecular AI, we are a cohesive group of accomplished deep learning researchers, software engineers, and innovators in drug discovery.Our mission is transformative: to develop next-generation AI foundation models that will enable new therapies for patients facing severe diseases. We engage in groundbreaking research at the convergence of machine learning, physics, and computational chemistry, continually challenging the limits of these disciplines.The Genesis AI team is at the helm of this revolution, creating extensive generative models that are trained on a wide array of molecular data, backed by robust computational infrastructure and simulation processes.This position integrates machine learning research, structural biology, and computational chemistry, necessitating rigorous technical proficiency and strong collaborative skills across disciplines.About the Role We are seeking an exceptional ML Research Engineer who excels at merging fundamental research with production-level engineering.As a vital member of the Genesis AI team, you will be instrumental in inventing, scaling, and deploying our next generation of foundation models for molecular science.You will collaborate closely with ML researchers, computational chemists, and drug discovery scientists to transform cutting-edge model concepts into operational systems that drive real drug discovery initiatives.Your responsibilities will include:Enhancing model pretraining pipelinesAdvancing reinforcement learning and post-training systemsOptimizing the performance of large molecular modelsImplementing structure prediction models like Pearl in production environments utilized by chemists and drug initiativesWe require a professional who can bridge the gap between ML and computational chemistry, facilitating swift transitions from research insights to practical applications.

Join Roblox as a Machine Learning Engineering Manager focused on Critical Harms where you will lead innovative projects that harness the power of machine learning to enhance safety and user experience on our platform. You will oversee a talented team dedicated to developing cutting-edge algorithms and models that proactively identify and mitigate risks.