"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, Python, JAX. Every listing is re-checked daily and closed roles are removed.

Showing 3 of 3 results

Member of Engineering (Post-training) Remote

Poolside AI Remote (EMEA/East Coast) Published 4 months ago

JAX Large Language Models PyTorch Transformers Python

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 →

Senior Machine Learning Engineer II, Ads Response Prediction Remote

Lead the design and development of core ML models for Instacart’s ads ecosystem in a fully remote role.

Instacart United States - Remote $201k–$253.5k/yr Published 3 months ago

pandas Spark JAX Machine Learning PyTorch

Flexible on stack

Apply →