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. Location This position is hybrid at our office in Emeryville, CA. Some travel may be required from time-to-time for in-person collaboration and work. Compensation The posted salary range is based on location in the Bay Area. The successful candidate will receive a competitive compensation package, commensurate with their experience and location.