"jax" Jobs

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

Showing 2 of 2 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 3 months ago

JAX Machine Learning Causal Reasoning representation learning reinforcement learning

Flexible on stack

Apply →

Applied Machine Learning Hybrid

Join Ideogram as an Applied ML Engineer to bridge research and product, turning generative models into production features.

Ideogram Toronto Published 2 months ago

diffusion JAX encoders PyTorch Transformers

Flexible on stack

Apply →