ONNX Jobs

Figure is an AI robotics company developing autonomous general-purpose humanoid robots with the goal of shipping robots with human-level intelligence. These robots are engineered to perform a variety of tasks in home and commercial markets. We are seeking a Staff AI Inference & Acceleration Engineer to join our Platform Software team. This role will be responsible for the on-board inference architecture for our humanoid robots, acting as the technical authority on how AI workloads are mapped, optimized, and executed across the robot's compute hardware. The primary objective is to reduce power consumption and cost while meeting strict latency and reliability demands for a real-time autonomous system.

$180k - $275k

CoreWeave is seeking a Principal Engineer to lead its Benchmarking & Performance team. This role will be responsible for a global performance data warehouse, ingesting, storing, transforming, and analyzing performance events across all data centers. You will also play a key part in industry-leading performance benchmarking publications, working closely with NVIDIA and the open-source community to demonstrate CoreWeave's performance and reliability leadership in AI.

$206k - $333k

Anduril Industries is a defense technology company focused on transforming military capabilities with advanced technology. We are seeking Senior Software Engineers to join our Tactical Recon & Strike team, which develops aerial small drones and associated equipment. This role will focus on solving complex perception problems for our UAVs, leveraging cutting-edge research in computer vision and machine learning. You will work on bringing products from concept to operational systems, collaborating with specialist teams to address critical customer needs. If you are passionate about creating powerful robotics platforms and tackling challenging technical problems, we encourage you to apply.

$220k - $292k

Anduril Industries is a defense technology company transforming military capabilities with advanced technology. We are seeking Software Engineers specializing in Computer Vision and Machine Learning Engineering to solve challenging perception problems for our UAVs. This role requires hands-on proficiency in C++ and/or Python within a Linux environment, leveraging cutting-edge research and deep expertise in areas like Object Detection and Object Tracking.

$191k - $253k

Anduril Industries is seeking a Machine Learning Engineer to join the Counter Intrusion MSE team. This role will own the machine learning stack for the team, focusing on designing and training multi-sensor object detection models for edge compute devices and developing learning algorithms to optimize autonomous systems. You will be responsible for the end-to-end design, implementation, and performance of the ML stack, including data collection and training infrastructure. Your work will directly contribute to Anduril's mission of delivering cutting-edge defense technology and will offer opportunities to identify and develop novel ML applications across the product portfolio.

$220k - $292k

We are seeking a Staff ML Tech Lead Manager to apply the latest research to solve our toughest problems. This role will be responsible for owning the entire machine learning stack for the Counter Intrusion team. You will design and train multi-sensor object detection models for perception on edge compute devices that push the boundaries of what's possible with our sensors. You will develop learning algorithms to optimize the behavior of autonomous systems. Your work will directly impact Anduril's ability to deliver cutting-edge defense technology, with the opportunity to identify and develop novel ML applications across our product portfolio.

$220k - $330k

You will help define how machine learning models run across Cloudflare’s global network, from frontier open LLMs and real-time voice models to customer-deployed models served on heterogeneous GPUs and next-generation accelerators. You’ll work with systems engineers, product teams, hardware partners, and AI/ML engineers to bring models into production with low latency, strong reliability, and efficient resource use. This role combines applied ML, inference optimization, evaluation, and production engineering, with a focus on benchmarking models, improving serving performance, validating quality, and building tooling that helps Cloudflare and its customers ship AI applications at Internet scale.

$194k - $266k

NVIDIA's TensorRT team is seeking talented engineers to advance high-performance AI inference on NVIDIA GPUs. This role focuses on developing software that accelerates state-of-the-art generative AI models for local and enterprise deployments. You will tackle challenging problems in GPU code generation, just-in-time compilation, and inference optimization to deliver high performance across diverse models and hardware configurations. As a Senior Software Engineer, you will design and implement advanced GPU inference software, translating a deep understanding of GPU execution into efficient code generation and measurable improvements in generative AI inference.

$152k - $288k

Confido is seeking a Senior ML Ops Engineer to be the first dedicated owner of its ML platform. This role will be responsible for the pipelines, serving, and cloud foundation that enable AI/ML models and agents to operate reliably and cost-efficiently in production. The engineer will ensure the platform can handle large-scale workloads, including document understanding, forecasting, and agentic systems, with a focus on making AI/ML development and deployment seamless for the AI/ML team.

$210k - $300k

Luma is seeking a Research Scientist/Engineer to focus on optimizing the performance of their multimodal models. This role involves profiling and enhancing GPU, CPU, and accelerator code to ensure efficient training and scalable deployment without compromising quality. You will be responsible for writing kernels and operations that maximize hardware utilization and minimize latency, requiring deep expertise in GPU optimization, transformer internals, and low-level programming.

$195k - $395k

Luma is seeking a Research Scientist/Engineer to focus on optimizing the performance of their multimodal models. This role involves profiling and enhancing GPU, CPU, and accelerator code to ensure efficient training and scalable deployment without compromising quality. You will be responsible for writing kernels and operations that maximize hardware utilization and minimize latency, requiring deep expertise in GPU optimization, transformer internals, and low-level programming.

At Mach9, ML infrastructure engineers build and maintain the systems that power production AI models for civil engineering and surveying. Our ML pipeline spans 10,000+ miles of labeled survey data, image segmentation networks, and 3D prediction models serving real-time inference to surveyors and engineers in the field. This role is ideal for mid-career ML infrastructure engineers with experience building for both training and inference. You'll build training pipelines that handle deep transformer models on hundreds of terabytes of 3D point cloud and image data. You'll also architect our inference infrastructure, delivering both heavy offline detection algorithms and real-time responsive inference that integrates directly with our CAD software.

$160k - $250k

At Mach9, ML Engineers build the perception models at the core of our AI-enabled CAD system, extracting 3D object and line features from dense LiDAR point clouds and imagery. This product-focused role involves working closely with customers to develop Computer Vision/Machine Learning pipelines and workflows that solve their problems. You will be responsible for shipping new features and product lines end-to-end, from data strategy and model training to evaluation. This position is ideal for early-career engineers who learn quickly, are curious, and find satisfaction in shipping impactful products.

$180k - $300k

Tavus is building the human layer of AI, aiming to make human-AI interaction as natural as face-to-face communication. We pioneer research in multimodal AI for modeling human communication (language, audio, video) and generating audio-visual avatar behavior, powering applications from text-to-video avatars to real-time conversational video. We are a Series B company backed by top investors, seeking individuals who thrive in startup environments, can prioritize independently, and are willing to take calculated risks to help pave the path forward.

Deepgram is seeking an Applied ML Engineer to focus on deploying their advanced speech models to a variety of edge devices and non-NVIDIA accelerators. This role involves adapting existing models to run efficiently within diverse hardware constraints, including different inference runtimes and operator sets. You will be responsible for optimizing models through techniques like quantization and precision adjustments, validating performance on real hardware, and building robust deployment pipelines for edge targets. This is a hands-on engineering position, distinct from research or cloud-focused roles, aimed at ensuring Deepgram's cutting-edge voice AI can be deployed effectively across a wide range of customer hardware.

$155k - $245k

Ambient.ai is seeking an AI Infrastructure Software Engineer to design, build, and optimize the AI infrastructure powering their real-time intelligence platform. This role involves working on systems to run state-of-the-art deep learning models, including computer vision, large language models (LLMs), large vision models (LVMs), and multimodal AI systems, across vast amounts of video data in real time. You will focus on scaling infrastructure for inference, evaluation, and continuous model improvement, partnering closely with research scientists and product engineering teams to bring AI advancements into production.

$168k - $205k

Parallel Wireless is seeking experienced wireless algorithm experts to research and design advanced cellular communication algorithms for 5G and beyond products. This role will leverage neural networks and machine learning techniques to reimagine mobile networks with innovative, energy-efficient Open RAN solutions. The ideal candidate will contribute to the future of telecommunications by driving innovation through green and sustainable networks.

NVIDIA AI for Media is seeking outstanding engineers to join its team, focusing on building and optimizing real-time AI solutions for media and entertainment workflows. This role will involve setting the technical direction for content authenticity AI, architecting model and pipeline strategies for detecting synthetic and manipulated media, and making critical build-vs-adapt decisions. The team works on core technologies to solve ambitious computer vision and deep learning problems, aiming to deliver studio-quality audio and high-resolution video enhancement and effects.

$272k - $431k