"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

AI Research Scientist - Spatial Reasoning

Join P-1 AI as an AI Research Scientist to advance spatial reasoning and physical world modeling in cutting-edge AI systems.

P-1 AI San Francisco Bay Area Published 1 month ago

JAX AI/ML PyTorch Python

Flexible on stack

Apply →

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 3 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 1 month ago

JAX PyTorch 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 Bay Area Published 8 months ago

JAX Machine Learning PyTorch TensorFlow Python

Flexible on stack 70% coding

Apply →

Machine Learning Scientist - Open Source Lead

Lead open-source research efforts at Arena Intelligence to advance AI model evaluation and transparency.

Arena Bay Area Published 8 months ago

JAX Machine Learning PyTorch TensorFlow Python

Flexible on stack

Apply →