Posted in

Aiml – Machine Learning Researcher – Mlr

Aiml – Machine Learning Researcher – Mlr

CompanyApple
LocationCupertino, CA, USA
Salary$143100 – $264200
TypeFull-Time
Degrees
Experience LevelJunior, Mid Level

Requirements

  • Demonstrated expertise in machine learning research working on at least some of the following topics – LLMs, Reasoning, Chain of Thought, Diffusion Language Models, Audio Generative Models, Speech Generative Models, and Speech Recognition Models, and 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.

Responsibilities

  • 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.
  • Provide technical mentorship and guidance.
  • Prepare technical reports for publication and conference talks.
  • Collaborate with broader teams across Apple.

Preferred Qualifications

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