Technical Staff Member - Embodied Agents

Join Moonlake, the forefront of AI technology for crafting immersive world simulations.Your Role: Design and train advanced embodied agents capable of perceiving their environment through vision, depth, and language; reasoning through memory and planning; and acting with precision in both continuous and discrete control.We are dedicated to fostering a collab…

Join Moonlake, a pioneering company at the forefront of artificial intelligence dedicated to crafting immersive world simulations.As a member of our Technical Staff, you will design and train advanced embodied agents capable of perception through vision, depth, and language, along with reasoning abilities including memory and planning, to execute actions through both continuous and discrete control methods.We pride ourselves on maintaining a collaborative and innovative team environment, currently operating from our San Mateo headquarters.

About Us:At Fireworks AI, we are at the forefront of developing innovative generative AI infrastructure. Our platform is recognized for delivering top-tier models and the industry's fastest, most scalable inference capabilities. As an industry leader in LLM inference speed, we are pushing boundaries with groundbreaking projects, including our own function calling and multimodal models. Fireworks is a Series C startup valued at $4 billion, supported by premier investors such as Benchmark, Sequoia, Lightspeed, Index, and Evantic. Our passionate and collaborative team is comprised of seasoned professionals from Meta PyTorch and Google Vertex AI.The Role: We are seeking a Training Infrastructure Engineer to design, build, and optimize the infrastructure that underpins our large-scale model training operations. Your contributions will be pivotal in establishing high-performance AI training infrastructure. You'll work closely with AI researchers and engineers to develop robust training pipelines, optimize distributed training workloads, and guarantee the reliability of model development.

Join us at Moonlake, where we harness the power of AI to create immersive world simulations.Model Development & ArchitectureDesign and refine innovative 2D/3D/image/video/audio diffusion architectures.Engage in conditioning tasks involving text, images, poses, layouts, and control signals with multi-modal encoders and guidance strategies.Training & OptimizationConduct large-scale diffusion training to enhance model performance.Focus on improving sample quality while optimizing computational resources through advanced objectives, distillation, and consistency models.Control & AlignmentImplement cutting-edge techniques such as ControlNet, LoRA, and IP-Adapters to manage style, identity, geometry, and control.Develop robust editing, inpainting, and personalization pipelines, including DreamBooth and custom subject/style tuning.Our vibrant team is dedicated to working on-site, currently located in San Mateo.

Join Fireworks AI as a Technical Staff Member specializing in Cloud Infrastructure. In this pivotal role, you will be at the forefront of our innovative cloud solutions, collaborating with a dynamic team of professionals dedicated to advancing cloud technologies. Your expertise will contribute to building and maintaining robust infrastructure that supports our evolving business needs.

Welcome to Moonlake, where we harness AI to create immersive world simulations.Our Mission: To enhance throughput, latency, and cost efficiency—deploying our models 2–10 times faster and more affordably, all while maintaining quality.Key Responsibilities:Optimize GPU performance through CUDA/Triton kernels, FlashAttention, paged attention, and CUDA Graphs.Manage the serving stack, including TensorRT-LLM/Triton Inference Server, vLLM/TGI; implement continuous batching and on-GPU KV reuse; explore speculative decoding/medusa and mixture-of-agents routing.Enhance parallelism via FSDP/ZeRO, TP/PP/expert parallel, and fine-tune NCCL.Implement quantization/PEFT techniques such as AWQ/GPTQ/FP8 and LoRA/DoRA serving.Oversee systems like Ray/k8s/Argo, ensuring observability with Prom/Grafana/OpenTelemetry, autoscaling, A/B testing infrastructure, and canary deployments with rollback capabilities.Ideal Candidate Profile:Candidates should have prior experience in infrastructure-heavy startups, particularly at companies like Databricks or Roblox. We are dedicated to fostering a collaborative, in-person team environment in San Mateo.

Welcome to Moonlake, where we leverage AI to craft immersive world simulations.Mission: Join us as an Applied AI Research Engineer focused on designing and coding intelligent agents (post-training and systems).Scope of Work:Design agentic systems: Develop tool catalogs, function calls, program synthesis, repair loops, and control mechanisms such as ReAct, Reflexion, ToT, and LangGraph, along with self-verification and sandboxed execution.Evaluation mindset: Create comprehensive task suites for multi-step coding, including full-stack LLM engineering, prompt libraries, routing, retrieval, KV-cache management, streaming, and telemetry.Security and isolation: Implement Docker/firejail, manage network egress controls, maintain secrets hygiene, and ensure dependency pinning for supply-chain integrity.Strong post-training capabilities: Conduct supervised fine-tuning, preference and trace reinforcement learning (DPO/RLAIF/RLHF), dataset curation, reward shaping, and safety filtering.Technical Signals:Experience shipping agents that successfully navigate real repository test suites from start to finish.Published research in the fields of agentic systems and code generation, contributing to frameworks or open-source evaluations such as LangGraph, AutoGen, Guidance, LEAP, and SWE-bench variants.Developed datasets from execution traces, demonstrating significant enhancements from data over parameters.We are committed to maintaining an on-site, collaborative team environment based in San Mateo.

About Us:At Fireworks AI, we are pioneering the next generation of generative AI infrastructure. Our innovative platform is designed to deliver the highest quality models with unparalleled speed and scalability in inference. Recognized as a leader in LLM inference speed, we continuously push the boundaries of technology through transformative projects, including our proprietary function calling and multimodal models. As a Series C company valued at $4 billion, we are supported by esteemed investors like Benchmark, Sequoia, Lightspeed, Index, and Evantic. Our team is a dynamic blend of visionaries and builders, with a strong foundation from Meta PyTorch and Google Vertex AI.The Role:As a Member of the Technical Staff specializing in Cluster Management at Fireworks AI, you will be pivotal in ensuring our world-scale virtual AI cloud operates reliably, efficiently, and at peak performance. Your expertise in large-scale distributed systems, cloud infrastructure, and operational excellence will be essential as you collaborate with top-tier software engineers and AI specialists to elevate our advanced AI platforms, addressing the rapid growth and evolving needs of our applications. This position is ideal for individuals passionate about building resilient, observable, and automated systems that drive customer success.

Join the innovative team at Moonlake, where we harness the power of AI to create real-time interactive content.Mission: Elevate performance metrics by enhancing throughput, reducing latency, and optimizing costs - deploying our models 2–10 times faster and at lower costs without compromising quality.Scope of Work:GPU Performance: Expertise in CUDA/Triton kernels, FlashAttention family, paged attention, and CUDA Graphs.Serving Stack: Proficiency with TensorRT-LLM/Triton Inference Server, vLLM/TGI; continuous batching; on-GPU KV reuse; speculative decoding/medusa; and mixture-of-agents routing.Parallelism: Experience with FSDP/ZeRO, TP/PP/expert parallel; NCCL tuning.Quantization/PEFT: Familiarity with AWQ/GPTQ/FP8; LoRA/DoRA serving.Systems: Knowledge of Ray/k8s/Argo, observability tools (Prom/Grafana/OpenTelemetry), autoscaling, A/B infrastructure, and canary + rollback.Tech Signals:Ideal candidates will have previous experience at infrastructure-heavy startups such as Databricks or Roblox.We are dedicated to maintaining an on-site, in-person team based in San Mateo.

Join Moonlake, where we harness the power of AI to craft immersive world simulations.Modeling & ArchitectureDevelop and enhance diffusion architectures for 2D/3D images, videos, and audio.Focus on conditioning techniques: text/image/pose/layout/control signals, multi-modal encoders, and guidance strategies.Training & OptimizationEngage in large-scale diffusion training processes.Enhance sample quality while managing compute trade-offs through improved objectives, distillation, and consistency models.Control & AlignmentUtilize techniques such as ControlNet, LoRA, and IP-Adapters for style, identity, geometry, and control adaptations.Design editing, inpainting, and personalization workflows, including DreamBooth and custom subject/style tuning.We are dedicated to maintaining an on-site, collaborative environment in San Mateo.

Join Moonlake, a pioneering company harnessing AI to develop immersive world simulations.Role OverviewEnhancing Training EfficiencyImplement data loaders, fusion techniques, activation rematerialization, and gradient checkpointing.Optimize training with FSDP/ZeRO/tensor+pipeline parallelism and NCCL tuning.Improving GPU and Kernel PerformanceConduct Nsight profiling, develop Triton/CUDA kernels, and create fused operations.Implement flash-attention style accelerations, sequence packing, and KV-cache optimizations.Optimizing InferenceFocus on low-latency serving, continuous batching, and speculative decoding strategies.Apply quantization methods (GPTQ/AWQ), distillation, and pruning techniques.Infrastructure and ReliabilityManage SLURM/Kubernetes multi-node jobs and ensure checkpoint hygiene.Maintain determinism, environment pinning, and effectively handle GPU failures.Our dedicated team thrives on collaboration in our San Mateo office.

Join us at Moonlake, where we leverage AI to craft immersive world simulations that push the boundaries of technology.Role OverviewEnhancing Training EfficiencyImplement advanced dataloaders, fusion techniques, activation rematerialization, and gradient checkpointing strategies.Utilize FSDP/ZeRO/tensor+pipeline parallelism and fine-tune NCCL settings for optimal performance.Boosting GPU and Kernel PerformanceConduct Nsight profiling and develop Triton/CUDA kernels along with fused operations.Implement flash-attention-style optimizations, sequence packing, and KV-cache improvements.Optimizing Inference ProcessesFacilitate low-latency serving, continuous batching, and speculative decoding techniques.Engage in quantization methods (GPTQ/AWQ), model distillation, and pruning practices.Infrastructure and Reliability EnhancementsManage SLURM/K8s multi-node jobs and ensure checkpoint hygiene.Focus on determinism, environment pinning, and effective GPU failure management.We pride ourselves on being an on-site, collaborative team located in San Mateo.

At Roblox, we empower millions of users to explore, create, play, learn, and connect through immersive 3D digital experiences crafted by our vibrant community of developers.Our mission is to redefine human interaction globally, enabling creativity and connection across devices. Join us as we strive to connect a billion people with optimism and civility, and play a vital role in shaping the future of digital interaction.As a Principal Machine Learning Engineer on the Creator Services Machine Intelligence team, you will lead the research and development of innovative Embodied AI and Behavioral Agents, transforming how games are conceptualized and played on Roblox. This role will bridge cutting-edge research with practical, large-scale applications, enabling sophisticated AI agents capable of dynamic 3D gameplay and enhancing user experience through features like automated playtesting and Smart NPCs that mimic human-like movement and strategic reasoning.In this position, you will engage with Imitation Learning (IL), Reinforcement Learning (RL), Computer Vision, Computer Graphics, Robotics, and Agentic Reasoning to develop adaptable agents that comprehend 3D environments, grasp game rules, formulate long-term strategies, and perform intricate physics-based actions.

Join our innovative team at Moonlake, where we harness the power of AI to create dynamic, real-time interactive content.Our Mission: To deliver exceptional product-level user experiences and full-stack solutions, transforming research insights into captivating, market-ready products swiftly.Key Qualifications:Exemplary product acumen; experience in rapid prototyping and ownership from inception to launch.Technical stack familiarity: Next.js/React, Tailwind/shadcn, edge/serverless architectures (Vercel/Cloudflare), tRPC/GraphQL, data synchronization, and authentication protocols.Ability to design and implement model endpoints, observability hooks, feature flags, and experimentation toggles.Technical Competencies:A portfolio showcasing polished AI applications in a production environment.Experience in developing and deploying design systems with a strong focus on performance metrics.We pride ourselves on being an on-site team, located in the vibrant city of San Mateo.

About Us:At Fireworks, we are at the forefront of generative AI infrastructure. Our platform stands out by delivering the highest-quality models with the fastest and most scalable inference capabilities in the industry. Recognized as a leader in LLM inference speed through independent benchmarking, we are spearheading innovation with projects like our proprietary function calling and multimodal models. As a Series C company valued at $4 billion and supported by leading investors such as Benchmark, Sequoia, Lightspeed, Index, and Evantic, we are a dynamic and collaborative team founded by veterans from Meta PyTorch and Google Vertex AI.We are in search of a Technical Staff Member for Evals & Post-Training Product to play a pivotal role in shaping how developers enhance models on our Fireworks platform. This position operates at the crossroads of product engineering, developer experience, and model quality.In this role, you will create products and workflows that establish a continuous connection between evaluation and post-training. You will facilitate internal teams in executing evaluations at scale, empower external developers through our open-source Eval Protocol SDK, and oversee critical product experiences aimed at fine-tuning custom models on Fireworks.Your work will span the entire stack—from APIs, SDKs, and backend systems to user-facing components in the web application—enabling users to easily create evaluations, interpret results, fine-tune models, and iterate swiftly. Additionally, you will engage directly with customers and internal teams to identify pain points, support real-world use cases, and transform recurring challenges into efficient product features.

Skydio stands at the forefront of the drone industry in the United States and is recognized globally for its cutting-edge autonomous flight technology, which is pivotal for the advancement of drones and aerial mobility. Our team merges profound knowledge in artificial intelligence with superior hardware and software development, operational excellence, and an unwavering commitment to customer satisfaction. We aim to enable a wider and more diverse audience of drone users, ranging from utility inspectors to first responders and military personnel in various scenarios.About the Role:We are seeking a Staff Technical Recruiter to play a vital role in assembling the teams that are defining the future of autonomous flight. In this position, you will drive essential engineering searches, collaborate directly with organizational leaders, and significantly influence our hiring processes as we expand. This senior, hands-on recruiting opportunity is perfect for an individual who excels in dynamic environments and possesses the skills to attract top-tier technical talent.PLEASE NOTE: Skydio operates under a hybrid work model, with teams working in-office three days a week. This position will be located at our headquarters in San Mateo, CA.Your Impact:Act as a strategic recruiting partner to senior engineering leaders, shaping organizational design, headcount planning, and recruitment strategies.Manage end-to-end talent acquisition strategies and execution for one or more of our core engineering divisions.Develop and fine-tune diverse, high-performing talent pipelines for specialized skill sets in emerging technology sectors.Lead executive-level recruitment efforts from initial discussions to successful closure, ensuring a top-notch candidate experience at every stage.Collaborate with People Operations, Compensation, and Finance to streamline headcount planning and recruitment processes for scalability and efficiency.Promote structured hiring methodologies, maintaining rigor and fairness in all recruitment decisions.Mentor fellow recruiters, sharing insights and best practices to enhance Skydio’s recruiting culture and continuously elevate our talent attraction and retention strategies.Your Qualifications:Extensive experience in recruiting within at least two of the following areas: Cloud, Mobile, Infrastructure, Embedded, Wireless, Camera/Optics, or Hardware Engineering.Demonstrated success in building technical teams at high-growth or impactful technology organizations.Strong ability to thrive in fast-paced environments and deliver results.

Join the innovative team at Moonlake, where we harness the power of AI to create real-time interactive content.Mission: As an Applied AI Research Engineer, you will focus on developing code agents that enhance our systems post-training.Scope of Work:Design agentic systems including tool catalogs, function calling, program synthesis/repair loops, and self-verification.Adopt an evaluation mindset by building comprehensive task suites for multi-step coding as a Full-stack LLM engineer, managing prompt libraries and telemetry.Ensure security and isolation through Docker/firejail, network egress controls, and secrets hygiene.Implement robust post-training strategies including supervised fine-tuning and dataset curation.Tech Signals:Demonstrated experience shipping agents that successfully navigate real-world repositories' test suites.Publication in the fields of agentic systems and code generation, or contributions to open-source frameworks like LangGraph or AutoGen.Ability to build datasets from execution traces and illustrate improvements driven by data.We are proud to maintain an on-site, in-person team located in San Mateo, CA.

Join Moonlake, a pioneering company harnessing AI to create immersive world simulations.Mission: Design exceptional product-level user experiences and deliver full-stack solutions, transforming research into captivating, shippable experiences with agility.Core Requirements:Demonstrated strong product taste with a knack for rapid prototyping and ownership from zero to one.Proficient in Next.js/React, Tailwind/shadcn, edge/serverless technologies (Vercel/Cloudflare), tRPC/GraphQL, and adept in file & data synchronization and authentication.Ability to design and implement model endpoints, observability hooks, feature flags, and experiment toggles effectively.Technical Expertise:A portfolio showcasing polished AI demos currently in production.Experience in building and shipping design systems, with a commitment to performance benchmarks.We prioritize an on-site, collaborative team environment based in San Mateo.

About Us:At Fireworks AI, we are revolutionizing the landscape of generative AI infrastructure. Our advanced platform provides the highest quality models with unparalleled speed and scalability in inference. We have been recognized as the industry leader in Large Language Model (LLM) inference speed and are committed to innovative projects, including our proprietary function calling and multimodal models. As a Series C company valued at $4 billion, we are backed by esteemed investors such as Benchmark, Sequoia, Lightspeed, Index, and Evantic. Our team consists of passionate builders, many of whom are veterans from Meta PyTorch and Google Vertex AI.The Role:As a vital contributor to our backend systems, you will help shape the foundation of Fireworks:Our primary web applicationModel and fine-tuning job orchestrationBilling and enterprise functionalitiesUser accounts, organization management, and access controlsPolicy enforcement and governanceAPIs and developer toolsAnd many exciting projects!This is platform engineering with tangible product impact. Your contributions will directly influence how our customers interact with AI. You will collaborate closely with product, frontend, infrastructure, and go-to-market teams to deliver comprehensive features rather than just completing tickets.

About Us:At Fireworks, we are pioneering the future of generative AI infrastructure. Our platform is recognized for delivering the highest-quality models with the industry's fastest and most scalable inference capabilities. With an independent benchmark as the leader in LLM inference speed, we are at the forefront of innovation, working on advanced projects including our proprietary function calling and multimodal models. As a Series C company valued at $4 billion, we are backed by esteemed investors such as Benchmark, Sequoia, Lightspeed, Index, and Evantic. Our team is made up of ambitious builders, including veterans from Meta PyTorch and Google Vertex AI.The RoleWe are seeking a skilled Data Platform Engineer specializing in Order-to-Cash (OTC) Revenue Transformation and AI Application Enablement. This role will involve taking ownership of and evolving the comprehensive billing, revenue, and business data pipeline—from usage metering and invoice generation to revenue recognition and financial reporting. You will be positioned at the nexus of Engineering, Finance, and Data, ensuring precise capture, billing, recognition, and reconciliation of every dollar generated across our five revenue streams.This impactful, cross-functional role requires hands-on engagement with our billing platform (e.g., Orb), accounting systems, data warehouse (BigQuery), and cloud marketplaces (AWS, GCP). Ultimately, you will contribute to the design of AI-enabled workflow agents that automate reconciliation, anomaly detection, and revenue operations once the core data infrastructure is fortified.