Machine Learning Researcher - Springtail

Responsibilities

• Hypothesize, test, and refine means of improving generalization performance of common architectural elements, including different forms of attention. This includes devising controlled datasets to elucidate e.g. learning order & learned representations.
• Think both mathematically and empirically about problems of runtime inference in gradient-trained networks, with an eye to the extensive literature on statistical learning and an open mind to the many forms of constrained optimization.
• Contribute to a well-documented and well-instrumented code base that is performant where necessary yet expeditious where experimental throughput demands.

Qualifications and Experience

• Masters or equivalent in machine learning, mathematics, or equivalent fields (strong candidates from neuroscience are encouraged to apply).
• Fluency with Pytorch, and familiarity with JAX, CUDA, and/or Triton + their open-source ecosystems.
• Demonstrated ability do fundamental research.
• Demonstrated ability to work in teams.

LocationThis position is hybrid at our office in Emeryville, CA. Some travel may be required from time-to-time for in-person collaboration and work.

CompensationThe posted salary range is based on location in the Bay Area.…