"jax" Jobs

6 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 6 of 6 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 Toronto, Ontario , Canada Published 3 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 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 →

Machine Learning Engineer Remote

Join Twilio as a Machine Learning Engineer to design AI-powered features that enhance customer conversations.

Twilio Remote - Spain Published 2 months ago

SpaCy NLTK JAX PyTorch TensorFlow

Flexible on stack

Apply →

Machine Learning Engineer Remote

Join Twilio as a Machine Learning Engineer to design AI-powered features that enhance customer conversations.

Twilio Remote - Ireland Published 2 months ago

SpaCy NLTK JAX PyTorch TensorFlow

Flexible on stack

Apply →