pytorch jobs in San Diego

Engineering Lead responsible for hands-on ML research, model development, and production AI system deployment in a live trading environment. You will work closely with quantitative research leadership to shape and execute the research agenda, design and train models, lead day-to-day technical execution across the engineering team, and mentor engineers to raise overall technical standards.…See more The role requires 4–6 years of applied ML/AI experience with 1–2 years of technical leadership, strong proficiency in Python and PyTorch, and a track record of shipping production ML systems where model performance directly impacts real-world outcomes.

Staff-level ML software engineer responsible for designing, implementing, and optimizing machine learning software for embedded systems including mobile, edge, automotive, and IoT products. Will develop and optimize ML kernels, frameworks (TensorFlow, PyTorch), and compilers in C++/Python to leverage specific hardware features on ARM, DSP, and GPU processors.…See more Responsibilities include performance optimization, integration, testing, tool development, and close collaboration with hardware teams. Requires a Master's degree in EE/CS or related field plus 3+ years of relevant experience.

Sr. SQA Engineer will design, develop, and execute comprehensive test plans for thick-client C# applications, web applications, cloud applications, and bioinformatics pipelines.…See more Responsibilities include functional, regression, integration, and performance testing on software interfacing with hardware systems; automating test cases using tools like Selenium and NUnit; identifying and tracking defects in Azure DevOps; and collaborating with software, mechanical, electrical, and bioinformatics engineers. Requires 6–8 years of SQA experience, preferably with genomics pipelines and cloud applications; strong proficiency in Windows, Linux, and Azure; and expertise with test automation frameworks.

Design, create, and implement production-grade AI/ML systems for Defense and Intelligence clients. You'll architect scalable machine learning solutions handling fast-moving data, deploy models across multiple data modalities using frameworks like TensorFlow and PyTorch, and integrate solutions into mission environments.…See more The role requires 2+ years of ML engineering or data science experience, proficiency with ML frameworks and programming, and hands-on work in deep learning, computer vision, NLP, or signal processing.

Staff-level machine learning software engineer responsible for designing, implementing, and optimizing ML software for embedded systems including mobile, edge, auto, and IoT products running on ARM, DSP, and GPU processors. Primary responsibilities include developing and optimizing ML kernels and frameworks (TensorFlow, PyTorch) for efficient hardware execution, implementing performance-critical software in C++/Python, and collaborating with hardware teams on joint design and optimization.…See more Will lead complex projects involving integration, testing, and tool development for embedded ML subsystems, coordinate cross-team dependencies, and contribute to design reviews and technical planning.

As a Machine Learning Engineer at Qualcomm, you will develop and optimize machine learning frameworks, tools, and algorithms for deployment across mobile, edge, automotive, and IoT products. You will collaborate with cross-functional teams to design hardware-software co-optimized solutions, build efficient inference and training systems, and create ML kernels and compiler tools.…See more The role requires proficiency in ML frameworks (TensorFlow, PyTorch, Keras), embedded systems optimization, and programming languages like Python or C++, with experience in problem domains such as NLP or multimedia.

Design, implement, and verify machine learning software for embedded ARM processors, DSPs, and GPUs in mobile, edge, automotive, and IoT products. Optimize ML kernels, frameworks (TensorFlow, PyTorch), and compilers to leverage specific hardware capabilities; develop in C++, C, or Python with focus on performance tuning and integration testing.…See more Collaborate with hardware teams on joint design/development, coordinate dependencies across teams, and contribute to design reviews and project meetings.

Qualcomm seeks a Senior AI Performance Architect to design hardware accelerator architectures optimized for AI training at scale. The role involves analyzing GPU/accelerator architectures, defining computational blocks supporting multiple datatypes and precisions, architecting memory and scale-out systems, and codesigning hardware with software/LLM requirements.…See more Responsibilities include competitive analysis, performance modeling, and pre-silicon prediction for ML training workloads. The ideal candidate spans software architecture, algorithm development, kernel optimization, and hardware accelerator design.

Design, build, and operate production ML systems across the full lifecycle—from data preparation and model training through deployment, monitoring, and continuous improvement. Partner with data scientists, software engineers, and platform teams to turn research prototypes into reliable, scalable services.…See more Work across recommendation, ranking, forecasting, classification, NLP, and generative AI depending on product priorities. Balance model quality, inference latency, scalability, and operational resilience as equally important outcomes.

Shield AI seeks software engineer interns for a 10–12 week paid program, working on autonomous systems across applications, embedded firmware, machine learning, and systems development. Interns will analyze requirements, write production-quality code in languages like C++, Python, or Java, and ship real features using Agile practices alongside full-time engineers.…See more The role spans user-facing applications, embedded flight software, perception models, and platform tools that support autonomy pipelines. Candidates should have a strong foundation in data structures, algorithms, version control (Git), and collaborative development.

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.

Qualcomm seeks an ML/computer vision engineer to develop and optimize AI-accelerated vision algorithms for edge devices (mobile, automotive, robotics, AR/VR). You will design algorithms for depth estimation, optical flow, video super-resolution, 3D reconstruction, and SLAM, then optimize deep learning models for resource-constrained SoCs.…See more The role spans algorithm research, model architecture design, performance profiling across memory/compute/power/bandwidth, and cross-functional collaboration with hardware and systems teams to ship production models for key customer segments.

Shield AI seeks software engineering interns for a 10–12 week paid program working on autonomous systems across applications, embedded firmware, machine learning, and systems platforms. Interns will analyze requirements, write production-quality code, and ship real features alongside full-time engineers using Agile practices.…See more Candidates should have proficiency in modern programming languages (C, C++, Python, Java, or JavaScript), solid foundations in data structures and algorithms, version control experience, and strong problem-solving skills. The role offers hands-on mentorship and potential for a full-time offer upon completion.

The Applied AI Engineer develops, deploys, and maintains generative AI solutions for insurance products and customer experiences. Responsibilities include implementing end-to-end AI/ML pipelines using AWS services (SageMaker, Lambda, ECS/EKS), feature engineering with Snowflake, model fine-tuning and testing, and monitoring production performance.…See more The role requires 3+ years of AI/ML experience with 1–2 years in generative AI or LLMs, hands-on cloud deployment experience, and proficiency in Python, PyTorch, TensorFlow, or Hugging Face Transformers.

Research Scientist Graduate role focused on designing and developing novel video codec algorithms for image and video compression across multiple formats (2D, multiview, 360-degree, point cloud, light field). Responsibilities include algorithm optimization, file format and transport protocol development, quality evaluation metrics, participation in standardization forums (MPEG, DASH, 3GPP, JVET, AVS), and publication of research papers and patent applications.…See more Requires a PhD in computer science or related field, strong fundamentals in algorithms and data structures, published research in top venues, and hands-on experience with C/C++ and multimedia compression standards.

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.

ByteDance seeks a Research Scientist Graduate with a PhD to work on multimedia standards, specifically image/video coding and processing. The role requires proficiency in Python and PyTorch for research and development.…See more This is an entry-level research position within ByteDance's Global Frontier Tech Recruitment Program, starting in 2027.

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