jax jobs in San Diego

As an Autonomy Engineer focused on VLA (Vision-Language-Action) pre-training, you will develop and train deep learning policies for humanoid robots, owning the full pipeline from data collection and curation to model deployment. You'll pre-train base models on diverse multi-embodiment trajectory corpora, fine-tune policies for specific tasks, work with MLOps teams to scale distributed training, and build continuous pipelines that ingest synthetic data and teleop logs.…See more This is primarily a deep learning role requiring 3+ years of experience shipping neural network models, hands-on expertise with LLMs, VLMs, or generative models, and strong proficiency in Python and PyTorch or JAX.

Senior Embedded Software Engineer at Qualcomm's Data Center Business Unit, developing software and firmware for cloud edge and data center ML acceleration on high-performance silicon. The role requires deep embedded systems expertise including BSP, device driver development, RTOS, secure boot, TrustZone, and low-level peripheral interfaces (I2C/SPI/UART), combined with knowledge of ML frameworks and data movement optimization.…See more You will work on critical technologies like ML acceleration, in-memory processing, memory management, and virtualization across cross-functional international teams. Requires 2+ years of low-level C programming, proficiency with GNU/LLVM tools, embedded Linux, JTAG/SWD debugging, and multi-threaded SoC architecture understanding.

Senior Embedded Software Engineer role focused on designing and developing software/firmware for cloud edge and data center ML accelerators on Qualcomm silicon. The engineer will work on ML acceleration, memory management, virtualization, and board support packages (BSP), device drivers, RTOS, secure boot, and peripheral interfaces (PCIe, I2C/SPI/UART, DMA).…See more Requires 2+ years of low-level C programming on embedded platforms with proficiency in BSP development, device driver work, Embedded Linux, RTOS, and debugging tools (JTAG/SWD). Preferred experience includes ML frameworks (PyTorch, JAX), QuRT RTOS, and Qualcomm-specific middleware.

Senior Embedded Software Engineer at Qualcomm's Data Center Business Unit designing and developing firmware and software solutions for cloud edge and data center ML workloads on high-performance silicon. The role requires deep expertise in embedded software (BSP, device drivers, RTOS, secure boot, TrustZone) and low-level C programming, with hands-on experience in board bring-up, debug tools (JTAG/SWD), and SoC architecture.…See more The candidate will work on critical technologies including ML acceleration, memory management, virtualization, and DMA-based data movement, collaborating across internal and cross-functional teams.

Design and improve autonomous AI trading systems deployed in financial markets, with core work on model architectures, training frameworks, and performance evaluation. You'll integrate novel data sources, optimize models for measurable accuracy gains and reduced training time, and build monitoring systems to detect anomalies in live trading strategies.…See more Requires strong ML fundamentals with hands-on experience in PyTorch, TensorFlow, or JAX, feature engineering, and model optimization; minimum 2 years of ML development experience preferred.

Lead AI software strategy and architecture for Qualcomm's automotive platform, directing deployment of computer vision, perception systems, LLMs, and generative AI workloads. Drive architectural decisions across multiple products and customer programs while mentoring cross-functional teams.…See more Ensure solutions meet automotive safety, performance, and reliability standards. Requires 8+ years of software/systems engineering experience with deep expertise in AI/ML, C/C++, and automotive software environments.

Principal Software Engineer leading AI software strategy and architecture for Qualcomm's automotive platform. Responsibilities include architecting and deploying advanced AI workloads (computer vision, LLMs, multimodal models, generative AI) on automotive systems, driving cross-functional technical decisions across AI, automotive engineering, and platform teams, mentoring engineers organization-wide, and ensuring solutions meet automotive safety, performance, and reliability requirements.…See more Requires 8+ years of software/systems engineering experience (or equivalent with advanced degree); preferred qualifications include 15+ years in AI/ML, embedded systems, or automotive software, expert C/C++ proficiency, and demonstrated experience with automotive safety standards (ISO 26262, ASPICE), edge AI inference, and Qualcomm AI frameworks.

Qualcomm's Data Center Business Unit seeks a Senior Embedded Software Engineer to design and develop software/firmware solutions for cloud edge and data center ML workloads on high-performance silicon. The role focuses on ML acceleration, memory management, virtualization, and board support packages (BSP), with deep expertise required in embedded Linux, device drivers, RTOS, secure boot, TrustZone, and low-speed peripherals (I2C/SPI/UART).…See more You will collaborate across internal and cross-functional teams to advance critical technologies including process-in-memory and efficient data movement via DMA. The position requires 2+ years of low-level C programming experience and a strong foundation in SoC architecture, multi-threaded programming, and embedded development tools.

Senior Embedded Software Engineer role at Qualcomm's Data Center Business Unit focused on designing and developing software/firmware for cloud edge and data center ML acceleration on high-performance silicon. The role requires deep expertise in embedded systems including board support packages, device drivers, RTOS, secure boot, TrustZone, and low-speed peripherals (I2C/SPI/UART), combined with proficiency in C/C++ development, debugging with JTAG/SWD, and embedded Linux.…See more Responsibilities include ML acceleration, process-in-memory optimization, memory management, virtualization, and collaboration with cross-functional teams. Ideal candidate has 2+ years of low-level C programming and background in SoC architecture, DMA, PCIe, and multi-threaded programming.

Lead Qualcomm's next-generation AI accelerator platform as senior engineering director, defining software vision, architecture, and roadmap while managing multiple engineering organizations spanning system software, device drivers, AI runtimes, compilers, ML frameworks, and platform SDKs. Partner with hardware teams on co-design, drive software readiness across chip development milestones, and establish performance/power/reliability goals.…See more Requires 9+ years software engineering experience (or PhD + 8 years), 4+ years with C/C++/Java/Python, and demonstrated expertise in AI/ML stacks, compiler technologies, runtime systems, or high-performance computing.

Sr. Director leading Qualcomm's next-generation AI accelerator platform software strategy, architecture, and delivery.…See more Manages multiple engineering organizations spanning system software, firmware, device drivers, AI runtimes, compilers, ML frameworks, SDKs, and validation. Defines performance and reliability goals, drives cross-functional hardware-software co-design, and partners with customers and hyperscalers. Requires 15+ years software engineering experience with 8+ years leading large, globally distributed teams; deep expertise in AI/ML stacks, compilers, runtime systems, or heterogeneous computing architectures.

Design and build large-scale multimodal foundation models that integrate high-dimensional biomedical imaging with molecular and language data. You will implement cross-domain data fusion strategies, develop high-performance ML pipelines for petabyte-scale datasets, and collaborate with scientists and engineers to translate biological complexity into scalable, production-ready code.…See more The role requires deep expertise in Computer Vision architectures, distributed training frameworks, and expert Python proficiency. You will contribute through high-impact publications or open-source work and demonstrated ability to handle the noise and sparsity inherent in biological data.