"jax" Jobs

5 open tech roles matching “jax”, taken straight from company career pages — not reposted from other job boards. Most in demand right now: JAX, PyTorch, Python. Every listing is re-checked daily and closed roles are removed.

Showing 5 of 5 results

Applied Research Intern, Proactive Intelligence & Customer World Models (PhD / Graduate Co-op) Remote

Join Block as an Applied Research Intern to develop proactive intelligence systems that enhance customer understanding and decision-making.

Block Bay Area, CA, United States of America Published 5 months ago

JAX Machine Learning Causal Reasoning representation learning reinforcement learning

Flexible on stack

Apply →

AI Research Scientist Visa

Join P-1 AI as an AI Research Scientist to push the boundaries of AI applied to the physical world in a hands-on role.

P-1 AI San Francisco Bay Area Published 3 months ago

JAX PyTorch Python

Flexible on stack

Apply →

Staff Software Engineer, Machine Learning (Consumer Revenue) Hybrid

Join Discord as a Staff Machine Learning Engineer to build ML systems for consumer revenue across various platforms.

Discord San Francisco Bay Area $272k–$340k/yr Published 1 month ago

JAX PyTorch TensorFlow Python

Flexible on stack

Apply →

Machine Learning Scientist - ML Research

Join Arena as a Machine Learning Scientist to evaluate AI models and contribute to impactful research in a collaborative environment.

Arena SF Bay Area Published 11 months ago

JAX Machine Learning PyTorch TensorFlow Python

Flexible on stack 70% coding

Apply →

Senior Machine Learning Engineer (Growth)

Join Discord as a Senior Machine Learning Engineer to build ML solutions that enhance user engagement and retention.

Discord San Francisco Bay Area $220k–$275k/yr Published 1 week ago

JAX PyTorch TensorFlow Python

Flexible on stack

Apply →