Hybrid Machine Learning Researcher

Posted 15 minutes ago

Apply now

About the role

  • Machine Learning Researcher at Astera Institute focusing on data-efficient and general model induction. Collaborating on innovative architectures with a focus on performance and throughput.

Responsibilities

  • Hypothesize, test, and refine means of improving generalization performance of common architectural elements
  • Think both mathematically and empirically about problems of runtime inference in gradient-trained networks
  • Contribute to a well-documented and well-instrumented code base that is performant where necessary yet expeditious where experimental throughput demands

Requirements

  • Masters or equivalent in machine learning, mathematics, or equivalent fields
  • Fluency with Pytorch
  • Familiarity with JAX, CUDA, and/or Triton + their open-source ecosystems
  • Demonstrated ability do fundamental research
  • Demonstrated ability to work in teams

Benefits

  • Competitive compensation package
  • Professional development opportunities
  • Well-documented and well-instrumented code base

Job title

Machine Learning Researcher

Job type

Experience level

Mid levelSenior

Salary

$150,000 - $300,000 per year

Degree requirement

Postgraduate Degree

Tech skills

Location requirements

Report this job

See something inaccurate? Let us know and we'll update the listing.

Report job