Aiml - Machine Learning Researcher, Mlr

Apple Apple · Big Tech · Cupertino, CA · Machine Learning and AI

Seeking junior and mid-level researchers for foundational ML research impacting Apple products. Role involves self-directed and collaborative research, publishing results, and providing technical mentorship. Requires expertise in ML research topics like RL, LLM training/adaptation, reasoning, diffusion models, audio, and multimodal models, with a strong publication record and deep learning toolkit experience.

What you'd actually do

  1. work on ambitious curiosity-driven long-term research projects that will impact the future of Apple, and our products
  2. work on innovative foundational research in machine learning
  3. advance the frontier of machine learning through a combination of self-directed research - proposing your own research ideas and demonstrating their feasibility, and collaborative research working with your colleagues on larger problems, sharing implementation and experimentation
  4. provide technical mentorship and guidance
  5. prepare technical reports for publication and conference talks

Skills

Required

  • Demonstrated expertise in machine learning research
  • Reinforcement Learning
  • LLMs training
  • LLMs test-time adaptation/scaling
  • Reasoning/Planning
  • Diffusion Language Models
  • Audio Generative/Recognition Models
  • Multimodal generative models
  • Publication record in relevant conferences (e.g., NeurIPS, ICML, ICLR, AAAI, CVPR, ICCV, ECCV, ACL, EMNLP, etc)
  • Hands-on experience working with deep learning toolkits such as Tensorflow or PyTorch
  • Strong mathematical skills in linear algebra and statistics

Nice to have

  • Ability to formulate a research problem, design, experiment, implement and communicate solutions
  • Ability to work in a diverse collaborative environment
  • PhD, or equivalent practical experience, in Computer Science, or related technical field
  • Hands on experience on at least some of the following: Torch Titan or Lingua, Sharded models, FSDP, DDP, fine-tuning pipelines for common models such as Llama, Mixtral, etc

What the JD emphasized

  • publication record in relevant conferences
  • regularly publish your results in the main relevant conference and journal venues
  • make sure that your research results are of high quality and reproducible

Other signals

  • curiosity-driven long-term research projects
  • innovative foundational research
  • advance the frontier of machine learning
  • self-directed research
  • collaborative research