DevJobs

Deep Learning Research Engineer - ML & Physics - Q

Overview
Skills
  • Python Python ꞏ 3y
  • PyTorch PyTorch ꞏ 3y
  • Custom loss functions
  • Regularization techniques
  • Generalization
  • Optimization landscapes
  • Neural network architectures
  • Model profiling
  • Transformers
  • Signal processing
  • Physics-informed neural networks
  • Neural operators
  • Attention
  • Implicit representations
  • GPU acceleration
  • FNO
  • Distributed training
  • DeepONet
  • Computational imaging
  • CNNs
Summary
Our team is new, growing, and evolving quickly, and we are hiring across multiple disciplines. We work at the intersection of machine learning, hardware, and software, creating technologies that reshape how people communicate and experience the world around them in everyday life.

Description
Apple designs consumer electronics that have touched millions and changed the way people interact with electronic devices around the world. The ML-Physics team works at the intersection of machine learning and physics, building models that leverage both data-driven learning and deep physical understanding to solve complex sensing and optical measurement challenges.

We are seeking a creative and hands-on Deep Learning Research Engineer to drive model and architecture innovation across our sensing platforms. In this role, you will go beyond off-the-shelf models, exploring novel neural network designs, designing rigorous training and validation pipelines, and translating inventive concepts into clean, high-performance implementations. While our domain deals with physical data, this is an ML-first role: we are looking for a deep learning specialist with strong research intuition, expertise in PyTorch, and the drive to solve non-standard, real-world problems.

Responsibilities
  • Conceive, design, and implement novel neural network architectures tailored for complex physical sensing, reconstruction, and inverse problems
  • Lead end-to-end hands-on research: data preparation, model formulation, training, hyperparameter exploration, and failure-mode analysis
  • Build, train, and rigorously validate deep learning models using Python and PyTorch
  • Formulate custom loss functions, regularization techniques, and inductive biases that capture domain-specific constraints
  • Design comprehensive validation and benchmarking suites to stress-test model robustness, generalization, and scaling behavior
  • Propose creative, non-conventional modeling ideas, stay abreast of state-of-the-art DL literature, and rapidly prototype proof-of-concepts
  • Collaborate cross-functionally with physicists and optical software engineers to benchmark models against simulated and experimental lab datasets

Minimum Qualifications
  • M.Sc. in Computer Science, Electrical Engineering, Data Science, Applied Mathematics, Physics, or a related technical field.
  • 3+ years of hands-on research experience in machine learning and deep learning (in industry R&D or an academic research lab)
  • Deep theoretical and practical understanding of modern deep learning fundamentals (e.g., neural network architectures, optimization landscapes, generalization, attention/transformers, CNNs, or implicit representations)
  • Proven hands-on proficiency in Python and PyTorch, including writing custom layers, loss functions, and efficient training loops
  • Demonstrated track record of creative problem-solving, proposing novel ideas, and executing them from hypothesis to experimental validation
  • Strong technical communication skills and fluency in English

Preferred Qualifications
  • Track record of research contributions (e.g., publications in top-tier conferences like CVPR, ICCV, NeurIPS, ICLR, ICML, or relevant patents)
  • Foundational understanding or interest in physics, wave optics, signal processing, or computational imaging
  • Experience with physics-informed neural networks (PINNs), neural operators (FNO/DeepONet), or integrating physical differential equations into loss functions.
  • Experience with large-scale distributed training, model profiling, and GPU acceleration

At Apple, we believe accessibility is a fundamental human right. You’ll find that idea reflected in everything here — in our culture, our benefits and our digital tools. By welcoming as many perspectives as possible, we help you build a career where you feel like you belong.

Learn about accessibility in Apple’s workplace

Role Number: 200692511-0865

Apple