Technical Staff Member - Diffusion Model

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 & Optimizat…

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.

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 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 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 collaborative, in-person team environment based in San Mateo, CA.

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.

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 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.

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 Genesis Molecular AI as a Staff Technical Recruiter and play a pivotal role in shaping our team by identifying and attracting top talent in the tech industry. You will be responsible for managing the end-to-end recruitment process, collaborating with hiring managers, and ensuring a smooth candidate experience.

Join Our Innovative TeamAt Genesis Molecular AI, we are a dedicated group of skilled deep learning researchers, software engineers, and pioneers in drug discovery. Our collective mission is to develop pioneering AI foundation models that will enable revolutionary therapies for patients suffering from severe conditions.We don't merely implement machine learning in biology; we engage in groundbreaking research at the convergence of machine learning, physics, and computational chemistry, challenging the limits of each discipline. As part of the Genesis AI team, you will collaborate with top-tier multidisciplinary researchers to design and construct generative and discriminative foundation models from an extensive range of molecular data, utilizing state-of-the-art computing resources and large-scale simulations.Your RoleThis position provides a unique platform for a scientific innovator to contribute to the future of generative AI in drug discovery. As an integral member of the Genesis AI team, you will guide and propel our research initiatives focusing on foundation models. You will spearhead significant research projects in areas such as reinforcement learning, innovative model architectures, and advanced pretraining and post-training techniques. Your primary goal will be to develop groundbreaking models and insights crucial for discovering new medications.This role demands a profound curiosity and a collaborative mindset. You will be an essential team player, working closely with our exceptional engineers and drug discovery specialists to transform complex concepts into reality. Additionally, we encourage you to be an influential voice in the scientific community, and we will actively support your efforts to publish research breakthroughs at premier ML conferences such as NeurIPS, ICML, and ICLR.Positions are available at several levels of seniority: Senior, Staff, and Principal.Key ResponsibilitiesLead transformative research projects from initiation to execution, addressing core challenges in generative modeling for molecular systems.Design and develop innovative models and algorithms, leveraging the latest advancements in diffusion models, flow matching, reinforcement learning, large language models, and other cutting-edge fields.

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.

About the TeamBecome a part of a pioneering team at the intersection of artificial intelligence and biochemistry.At Genesis Molecular AI, we are a dynamic collective of distinguished deep learning researchers, software engineers, and innovators in drug discovery. Our ambitious mission is to develop the next frontier of AI foundation models to unlock revolutionary treatments for patients battling serious diseases.We are not merely applying machine learning to biological problems; we are engaged in groundbreaking research that bridges machine learning, physics, and computational chemistry, challenging the limits of each discipline. The Genesis AI team is constructing the engine for this transformation. You will collaborate closely with leading multidisciplinary researchers to design and implement generative and discriminative foundation models at scale, utilizing extensive computational resources and large-scale simulations.About the RoleThis is a unique opportunity for a scientific innovator to influence the future landscape of generative AI in drug discovery. As a pivotal member of the Genesis AI team, you will direct our research strategy for foundation models. You will spearhead critical research initiatives in areas such as reinforcement learning, innovative model architectures, and advanced pretraining and post-training techniques. Your primary goal will be to develop transformative models and insights that play a crucial role in the discovery of new medications.This position demands a profound curiosity and a commitment to collaboration. You will be an integral team player, partnering closely with our exceptional engineers and drug discovery specialists to realize complex concepts. Additionally, we encourage you to engage with the scientific community, and we will actively support your endeavors to publish significant research findings in prestigious ML conferences such as NeurIPS, ICML, and ICLR.We are hiring for various levels of seniority: Senior, Staff, and Principal.Your ResponsibilitiesLead innovative research projects from inception to execution, addressing fundamental challenges in generative modeling for molecular systems.Design and develop advanced models and algorithms, leveraging the latest advancements in diffusion models, flow matching, reinforcement learning, and large language models.Conduct large-scale experiments utilizing our premier computational resources to validate your research.

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.