"jax" Jobs

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

Showing 3 of 3 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 Toronto, Ontario , Canada Published 2 months ago

JAX Machine Learning Causal Reasoning representation learning reinforcement learning

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 2 months ago

JAX Machine Learning Causal Reasoning representation learning reinforcement learning

Flexible on stack

Apply →

Staff Machine Learning Engineer (Modeling), Support

Lead the development of machine learning initiatives to enhance customer support experiences at Block.

Block Seattle, WA, United States of America $276.8k–$415.2k/yr Published 2 months ago

JAX PyTorch TensorFlow Python

Flexible on stack

Apply →