Junior ML Scientist (AI)

Location: San Francisco, United States; on-site

Salary: $75,000–$200,000 per year, plus equity; compensation varies with experience.

Company

hirify.global develops hardware that accelerates probabilistic inference and enables new approaches to training models in the thermodynamic paradigm.

What you will do

• Collaborate with senior researchers to derive theories for probabilistic models and learning rules, including energy-based and diffusion models.
• Scale experimentation infrastructure and optimize the design space of machine learning models.
• Implement, visualize, and evaluate architectures, training algorithms, and benchmarks.
• Publish research papers, contribute to open source, and communicate design insights to the hardware team.

Requirements

• Experience with scientific Python and JAX or a comparable deep learning framework such as PyTorch, TensorFlow, or Keras.
• Strong foundations in probability and linear algebra.
• Projects or papers demonstrating hands-on experience in applied machine learning and data science.
• Employment requires authorization to receive information and controlled items subject to U.S. export control laws and regulations.

Nice to have

• Knowledge of deep learning theory, over-parameterization, and scaling laws.
• Experience with energy-based models, diffusion models, graph neural networks, or graph message passing algorithms.
• Experience with experimentation and training infrastructure such as Slurm, Ray, Kubernetes, or Weights & Biases.
• Background in information geometry, computational Bayesian methods, MCMC, or variational inference.
• Publications in leading machine learning conferences.

Culture & Benefits

• Flexible residency program available on a part-time or full-time basis.
• Minimum program duration of three months.
• Greater autonomy than a typical internship.
• Opportunities to publish papers and contribute to open source.
• Salary and equity compensation vary with experience.