Talent.com
Research Scientist - Machine Learning
Research Scientist - Machine LearningExtropic • San Francisco, CA, US
Research Scientist - Machine Learning

Research Scientist - Machine Learning

Extropic • San Francisco, CA, US
7 days ago
Job type
  • Full-time
Job description

Job Description

Job Description

Overview

Extropic’s hardware massively accelerates certain kinds of probabilistic inference.  Our ML team works on the science of training models in the thermodynamic paradigm, and we are looking for senior research and engineering talent to derive probabilistic ML theory, empirically demonstrate its scaling properties, and deploy performant models. Senior hires will be leading their own research direction and are therefore expected to quickly become experts across our abstraction stack, including the hardware, software, physics, and math.

Responsibilities

  • Collaborate with senior researchers, residents, engineers, and physicists to derive the theory of new probabilistic models and their learning rules, including energy-based models and diffusion models.
  • Scale up experimentation infrastructure and optimize over the design space of models.
  • Implement, visualize, and evaluate new architectures, training algorithms, and benchmarks.
  • Publish papers, contribute to open source, and communicate design insights to our hardware team.
  • Create production models for domain experts using customer data.

Required Qualifications

  • Experience in scientific Python and at least one deep learning framework (PyTorch, JAX, TensorFlow, Keras)
  • Extremely strong foundations in probability and linear algebra
  • Familiarity with deep learning theory and literature, including theory of over-parameterization and scaling laws
  • Publications in top ML conferences (NeurIPS, ICML, ICLR, CVPR)
  • Experience training high-performance models, including familiarity with infrastructure (Slurm, Ray, Weights & Biases)
  • Experience deploying models, including familiarity with infrastructure (Ray, AWS, ONNX)
  • Preferred Qualifications

  • Experience designing probabilistic graphical models (PGM)
  • Experience training energy-based models (EBMs) or diffusion models
  • Experience with numerical methods in diffeq solvers
  • Experience with message passing or training graph neural networks (GNNs)
  • Strong theoretical background in information geometry
  • Strong theoretical background in random matrix theory
  • Strong grasp of computational Bayesian methods, including MCMC sampling methods and variational inference
  • Salary and equity compensation will vary with experience

    Extropic is an equal opportunity employer

    We may use artificial intelligence (AI) tools to support parts of the hiring process, such as reviewing applications, analyzing resumes, or assessing responses. These tools assist our recruitment team but do not replace human judgment. Final hiring decisions are ultimately made by humans. If you would like more information about how your data is processed, please contact us.

    Create a job alert for this search

    Research Scientist • San Francisco, CA, US

    Related jobs
    Machine Learning Research Scientist / Engineer, Reasoning

    Machine Learning Research Scientist / Engineer, Reasoning

    Scale AI, Inc. • San Francisco, CA, United States
    Full-time
    At Scale AI, our mission is to accelerate the development of AI applications.For 8 years, Scale has been the leading AI data foundry, fueling the most exciting advancements in AI, including generat...Show more
    Last updated: 30+ days ago • Promoted
    Sr. Staff Machine Learning Engineer, Applied Research Science

    Sr. Staff Machine Learning Engineer, Applied Research Science

    Pinterest • San Francisco, CA, United States
    Full-time
    Millions of people around the world come to our platform to find creative ideas, dream about new possibilities and plan for memories that will last a lifetime. At Pinterest, we're on a mission to br...Show more
    Last updated: 1 day ago • Promoted
    Senior Machine Learning Manager

    Senior Machine Learning Manager

    VirtualVocations • Concord, California, United States
    Full-time
    A company is looking for a Senior Machine Learning Manager, Gen AI & Knowledge AI.Key Responsibilities Lead the vision, design, and execution of LLM-powered AI products, including system architec...Show more
    Last updated: 30+ days ago • Promoted
    Machine Learning Engineering Manager

    Machine Learning Engineering Manager

    VirtualVocations • Concord, California, United States
    Full-time
    A company is looking for a Manager, ML Engineering to lead efforts in Accelerated Machine Learning tools and Analytics Pipelines on GPUs. Key Responsibilities Lead, mentor, and grow the engineerin...Show more
    Last updated: 30+ days ago • Promoted
    Tech Lead / Manager, Machine Learning Research Scientist- LLM Evals

    Tech Lead / Manager, Machine Learning Research Scientist- LLM Evals

    Scale AI, Inc. • San Francisco, CA, United States
    Full-time
    As the leading data and evaluation partner for frontier AI companies, Scale is dedicated to advancing the evaluation and benchmarking of large language models (LLMs). We are building industry-leadin...Show more
    Last updated: 11 hours ago • Promoted • New!
    Machine Learning Research Engineer - Robotics

    Machine Learning Research Engineer - Robotics

    Scale AI, Inc. • San Francisco, CA, United States
    Full-time
    Scale's Robotics business unit is dedicated to solving the data bottleneck in Physical AI.This position will be a key contributor in conducting applied research in Robotics and developing ML pipeli...Show more
    Last updated: 30+ days ago • Promoted
    Machine Learning Research Engineer, Enterprise ML Systems

    Machine Learning Research Engineer, Enterprise ML Systems

    Scale AI, Inc. • San Francisco, CA, United States
    Full-time
    AI is becoming vitally important in every function of our society.At Scale, our mission is to accelerate the development of AI applications. For 9 years, Scale has been the leading AI data foundry, ...Show more
    Last updated: 3 days ago • Promoted
    Remote Machine Learning Engineer

    Remote Machine Learning Engineer

    VirtualVocations • Concord, California, United States
    Remote
    Full-time
    A company is looking for a Remote Machine Learning Engineer.Key Responsibilities Design, develop, and architect high-performance AI / ML services and pipelines in a cloud-based environment Contrib...Show more
    Last updated: 3 days ago • Promoted
    Staff Machine Learning Engineer

    Staff Machine Learning Engineer

    VirtualVocations • Concord, California, United States
    Full-time
    A company is looking for a Staff Machine Learning Engineer - Wildfire.Key Responsibilities Architect and build advanced ML models to predict vegetation and fuel conditions Design and maintain da...Show more
    Last updated: 30+ days ago • Promoted
    Senior AI / ML Scientist

    Senior AI / ML Scientist

    VirtualVocations • Concord, California, United States
    Full-time
    A company is looking for a Senior AI / ML Applied Scientist.Key Responsibilities Investigate and evaluate the latest LLM and MLM models for healthcare applications Develop data pipelines for ML tr...Show more
    Last updated: 30+ days ago • Promoted
    Senior Research Scientist

    Senior Research Scientist

    VirtualVocations • Concord, California, United States
    Full-time
    A company is looking for a Senior Research Scientist - Digital Biology.Key Responsibilities Developing, implementing, benchmarking, and accelerating multimodal models in digital biology and drug ...Show more
    Last updated: 30+ days ago • Promoted
    Sr Machine Learning Engineer, Applied Research Science

    Sr Machine Learning Engineer, Applied Research Science

    Pinterest • Palo Alto, CA, United States
    Full-time
    Millions of people around the world come to our platform to find creative ideas, dream about new possibilities and plan for memories that will last a lifetime. At Pinterest, we're on a mission to br...Show more
    Last updated: 30+ days ago • Promoted
    Senior UX Researcher - AI / ML

    Senior UX Researcher - AI / ML

    VirtualVocations • Concord, California, United States
    Full-time
    A company is looking for a Senior UX Researcher (AI / ML - Healthcare / SaaS).Key Responsibilities Drive AI-enhanced product insights through user behavior analysis and feedback Leverage continuous ...Show more
    Last updated: 9 hours ago • Promoted • New!
    Machine Learning Research Scientist / Research Engineer, Post-Training

    Machine Learning Research Scientist / Research Engineer, Post-Training

    Scale AI, Inc. • San Francisco, CA, United States
    Full-time
    Scale works with the industry's leading AI labs to provide high quality data and accelerate progress in GenAI research.We are looking for Research Scientists and Research Engineers with expertise i...Show more
    Last updated: 30+ days ago • Promoted
    Senior Machine Learning Engineer

    Senior Machine Learning Engineer

    Meltwater • Redwood City, CA, United States
    Full-time
    Meltwater's Consumer Intelligence AI Team is looking for a.Natural Language Processing or Computer Vision features relying on the literature's state of the art. Those features are meant to be integr...Show more
    Last updated: 21 days ago • Promoted
    Senior Machine Learning Engineer

    Senior Machine Learning Engineer

    VirtualVocations • Concord, California, United States
    Full-time
    A company is looking for a Senior Machine Learning Engineer, Security.Key Responsibilities Design, build, and deploy machine learning models to detect and mitigate security threats Develop algor...Show more
    Last updated: 30+ days ago • Promoted
    Machine Learning Research Lead, Security & Policy Research Lab

    Machine Learning Research Lead, Security & Policy Research Lab

    Scale AI, Inc. • San Francisco, CA, United States
    Full-time
    As the leading data and evaluation partner for frontier AI companies, Scale plays an integral role in understanding the capabilities and safeguarding AI models and systems.Building on this expertis...Show more
    Last updated: 14 days ago • Promoted
    Machine Learning Ops Engineer

    Machine Learning Ops Engineer

    VirtualVocations • Fremont, California, United States
    Full-time
    A company is looking for a Machine Learning Ops Engineer.Key Responsibilities Design and implement AI cost tracking and observability frameworks across Azure, Google Cloud, and Snowflake Collabo...Show more
    Last updated: 30+ days ago • Promoted