pytorch jobs in San Diego

Principal-level individual contributor role leading AI/ML architecture and engineering strategy at scale. Requires 13+ years across ML, data engineering, and distributed systems, with 3+ years shipping production Generative/Agentic AI systems.…See more Hands-on expertise in RAG, vector databases, LLM optimization, agentic orchestration, and MLOps/LLMOps practice. Will set technical direction, own architecture decisions, drive platform initiatives, and raise engineering standards across the organization without direct people-management responsibility.

Engineering Lead responsible for designing, developing, and deploying production ML models for live trading environments while leading day-to-day technical execution across the engineering team. You will work with leadership to shape the research agenda, mentor engineers, drive adoption of AI tools to accelerate engineering workflows, and build/improve core ML infrastructure including data pipelines, training workflows, and inference systems.…See more The role requires 4–6 years of applied ML/AI experience with 1–2 years leading technical projects or teams, strong proficiency in Python and PyTorch, and a track record of shipping ML systems where model performance directly impacts business outcomes.

Lead end-to-end model optimization and transformation for large language models, vision language models, diffusion, and multimodal models on Qualcomm inference accelerators. Architect PyTorch-based optimization strategies, drive graph capture and deployment using PyTorch/ONNX/torch.compile, and design fusion kernels using Triton or similar DSLs.…See more Partner with compiler, performance, and accuracy teams to co-design lowering strategies, optimize transformer-specific patterns (KVcache, decoding, long context), and scale distributed inference across multi-core and multi-device systems. Requires expert-level PyTorch proficiency, deep transformer architecture knowledge, hands-on experience with torch.compile/TorchDynamo, and strong foundation in ML accelerators and distributed systems.

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.

Software Engineering Intern III position for Summer 2027 at BAE Systems, supporting defense and aerospace projects. Interns will work alongside experienced engineers on impactful projects using modern full-stack technologies including React/Angular, AWS/Azure/Kubernetes, Spring, and various databases (SQL and NoSQL).…See more The role requires proficiency in Java, C++, C#, Python, or JavaScript, strong fundamentals in data structures and object-oriented programming, and a drive to learn. Participation in BAE Systems' LEAP Internship Program includes career development, mentorship, and networking opportunities.

Design, architect, and deploy on-device AI prototype software for edge computing applications at Qualcomm's AI Research division. You will work with modern deep learning frameworks (PyTorch), optimize neural network models for embedded deployment, and integrate cutting-edge AI technologies into mobile, IoT, and autonomous systems.…See more The role requires strong Python and C/C++ skills, deep learning expertise, embedded software development experience, and solid software engineering fundamentals.

Design and develop Qualcomm's in-house robotics simulation platform and AI infrastructure, integrating modern 3D engines, GPU-accelerated compute, and machine learning to enable robot policy training and validation. Build digital twins, synthetic data pipelines, and hardware-in-the-loop configurations that bridge simulation and real Qualcomm robotics hardware.…See more Partner with robotics researchers to iterate on simulation fidelity, sensor modeling, and continuous learning workflows that move algorithms from simulation to deployed robots. This role demands depth in robotics simulators (MuJoCo, Isaac Sim, Unreal/Unity), GPU programming (CUDA), reinforcement learning at scale, and AI infrastructure orchestration (Kubernetes, Argo Workflows).

Develop high-performance GPU kernels and runtimes for accelerating machine learning and image processing on Qualcomm Adreno GPUs. Optimize drivers for OpenCL, OpenGL ES, Vulkan, and DirectX APIs, working closely with hardware designers to understand and improve hardware/software interfaces.…See more Implement machine learning inference and training acceleration, write unit tests and documentation, and collaborate with partners and OEMs. Requires strong C/C++ skills, familiarity with GPGPU APIs like CUDA or OpenCL, and understanding of data structures, algorithms, and object-oriented programming.

Sr. Staff Software Engineer role designing and developing Multimedia, AI, and Gen AI SDKs/framework components for IoT products including drones, cameras, and media devices.…See more Requires 6+ years of software engineering experience with 3+ years in C, C++, Java, or Python, with strong preference for deep embedded systems expertise, real-time solutions, and AI/ML inference frameworks. Responsibilities include leading architecture and development of GStreamer-based SDKs, driving feature design across diverse product categories, and providing technical leadership on code quality and cross-functional collaboration.

Senior ML engineer role focused on architecting, designing, and deploying on-device AI prototype software at the edge. Requires strong Python and C/C++ skills, deep learning expertise (PyTorch), and embedded software development experience.…See more The role involves model fine-tuning, quantization, hardware acceleration, and inference optimization for deployment on smartphones, autonomous vehicles, IoT, and robotics platforms. Must have 3+ years relevant experience and a strong theoretical ML background combined with modern software engineering practices.

Senior Staff Engineer responsible for designing and building an MLOps platform that supports distributed AI training, reinforcement learning, and foundation model development at scale. You will architect Kubernetes-native infrastructure for GPU workloads, design self-service AI development workflows, manage the data and model lifecycle, and lead platform distribution across cloud, on-premises, and air-gapped environments.…See more The role requires deep expertise in modern AI frameworks (PyTorch, Hugging Face Transformers), distributed systems, GPU scheduling, and cloud-native infrastructure, with a focus on enabling researchers and engineers to move from experimentation to production rapidly.

Staff Machine Learning Engineer role focused on designing and maintaining Qualcomm AI Hub's end-to-end ML compilation pipeline, from PyTorch and ONNX model ingestion through graph optimization to deployable artifacts on Snapdragon SoCs. Responsibilities span compiler infrastructure (graph transformation, op validation, backend dispatch across CPU/GPU/NPU), model catalog automation and CI/CD, and developer tooling for profiling and debugging.…See more Requires 4+ years of software/systems engineering experience (or 3+ with a master's, or 2+ with a PhD), with preferred expertise in ML compiler concepts, ONNX/PyTorch export workflows, and on-device deployment frameworks.

Lead ML research and production model development for a quantitative trading firm, working hands-on to design, train, and deploy ML systems that directly impact live trading environments. You will drive technical execution across the engineering team, mentor engineers to raise the bar, and translate ambiguous research problems into executable modeling work with measurable real-world impact.…See more The role requires 4–6 years of applied ML experience with 1–2 years of technical leadership, strong fluency in Python and PyTorch, and a track record of shipping production systems in high-stakes contexts.

Join Booz Allen Hamilton as an AI Software Developer Intern supporting U.S. Navy missions in San Diego.…See more You'll write and debug code (primarily Python), contribute to AI/ML-enabled automation projects, and work on software integration tasks alongside experienced engineers and cybersecurity specialists. The role is structured to support your academic commitments while you pursue a bachelor's degree in computer science, cybersecurity, or a related STEM field by Summer 2027.

Lead the design, development, and operation of Qualcomm's enterprise AI platform, supporting agentic AI, model lifecycle management, and multi-cloud ML/inference infrastructure. Own core platform services including identity/RBAC, secrets, service meshes, observability, vector stores, and model gateways across on-prem GPU clusters and managed cloud services.…See more Manage a ~10-engineer global team (platform, SRE, MLOps/LLMOps), drive incident response and continuous improvement, and partner with product and security teams on AI governance and high-impact use cases.

Principal-level AI/ML Platform Architect responsible for designing and optimizing AI/ML systems across Qualcomm's hardware and software stacks, particularly for Windows PC/Laptop platforms. The role requires deep expertise in performance optimization, bottleneck analysis, and end-to-end system characterization across GPUs, NPUs, and multiple AI domains (computer vision, audio, generative AI, agentic AI).…See more Candidates must demonstrate 15+ years in high-performance computing systems engineering with 5+ years specifically in AI system optimization, and will interface with cross-functional teams to deliver reference implementations, tools, and documentation for third-party developers.

The Senior Machine Learning Engineer will design and develop advanced ML models for automotive radar signal processing, including end-to-end pipelines from data preprocessing through model training and optimization. The role requires expertise in deep learning architectures (CNNs, RNNs, Transformers), radar signal analysis, and deploying models for real-time embedded systems with constraints on latency, memory, and power.…See more Responsibilities include evaluating state-of-the-art ML techniques, optimizing models for production, and translating research into scalable algorithms. The position demands a Master's or PhD in a quantitative field, 3+ years of ML research experience, strong proficiency in Python and frameworks like PyTorch or TensorFlow, and foundational knowledge of radar fundamentals.

Design and develop advanced machine learning models for automotive radar signal processing, including end-to-end ML pipelines from data preprocessing through model training and optimization. Evaluate novel deep learning architectures (CNNs, RNNs, Transformers) applied to radar data in time-domain, frequency-domain, and range–Doppler representations.…See more Optimize models for real-time embedded deployment under latency, memory, and power constraints, and translate research outcomes into scalable, production-ready algorithms. Requires a Master's or PhD in a quantitative field, 3+ years of ML research experience, strong radar and signal processing fundamentals, and proficiency in Python, PyTorch/TensorFlow, and cloud platforms.

Staff/Sr. Staff Software Engineer role focused on AI inference optimization on Snapdragon platforms, including model optimization, quantization, graph transformations, and runtime execution for LLMs, LVMs, and LMMs.…See more You will design and implement graph lowering and optimization techniques within ONNX Runtime, ExecuTorch, and Qualcomm AI Stack SDK, working across ML algorithms, inference systems, and hardware integration. The role requires 6–8+ years of software development experience, 3+ years in AI/ML inference or model optimization, deep expertise in Python and C/C++, PyTorch/ONNX, and transformer architectures. You will mentor junior engineers, drive features end-to-end, and collaborate across ML Research, hardware, product, and QA teams.

Design and develop embedded and cloud edge software applications focused on AI/GenAI, multimedia, and computer vision SDKs and frameworks for IoT products including drones, cameras, and AI boxes. Collaborate with systems, hardware, and architecture teams to build high-performance, real-time embedded software solutions.…See more Lead GenAI LLM/VLM inference workflows, multi-stream AI pipelines, and multimedia frameworks across Android, Linux, and embedded systems. Requires 2+ years of software engineering experience with C/C++, Python, or Java, plus preferred expertise in PyTorch, TensorFlow, ONNX Runtime, LangChain, LlamaIndex, Android architecture, Linux system programming, and graphics/GPU pipelines.

Qualcomm AI Research seeks a Senior AI Research Quantization Engineer to develop algorithms for efficient generative AI, LLMs, and multimodal models optimized for on-device deployment. The role focuses on advanced quantization techniques, model compression, inference optimization (batching, KV caching, speculative decoding), and system prototyping using Python and PyTorch.…See more You will collaborate across hardware, software, and systems teams to enable state-of-the-art models to run on power- and memory-constrained devices including smartphones, autonomous vehicles, robotics, and IoT platforms. A Bachelor's degree plus 2+ years of related engineering experience (or Master's with 1+ year, or PhD) is required.

Develop software for the Qualcomm AI Stack SDKs (QAIRT and Genie) to enable execution of generative AI models and large language models on Snapdragon platforms. You will validate, optimize, and debug machine learning inference solutions, collaborating with cross-functional teams on feature development, performance analysis, and system reliability.…See more The role requires expertise in machine learning frameworks, embedded systems optimization, programming languages like Python or C++, and low-level OS/hardware interactions.

Build and optimize scalable LLM inference platforms at Qualcomm's Cloud AI team, implementing advanced serving techniques like KV-cache management, speculative algorithms, and model optimization. Contribute to production serving frameworks (vLLM, SGLang, Triton, TGI) and work with customers on deployment solutions.…See more Collaborate with compiler, firmware, and platform teams to drive efficient serving through autoscaling, load balancing, and routing. Requires deep understanding of transformer architectures, strong PyTorch and Python skills, computer architecture knowledge, and hands-on experience profiling and optimizing deep learning workloads.

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.

As a Machine Learning Application Developer, you will develop and optimize neural rendering applications and ML software for efficient AI execution on Samsung's mobile GPUs. Working with hardware and software teams, you'll optimize performance and resource utilization by developing API-level software, ML operator implementations (GEMM, convolution, activations), and conducting low-level performance analysis.…See more The role requires strong C/C++/Python skills, proficiency with Vulkan/OpenGL/OpenCL, hands-on neural rendering experience, and deep understanding of GPU architecture and performance optimization techniques.