"jax" Jobs

7 open tech roles matching “jax”, taken straight from company career pages — not reposted from other job boards. Every listing is re-checked daily and closed roles are removed.

No "jax" jobs in New York City Possible For Exceptional Candidates right now — showing "jax" jobs in all locations.

Showing 7 of 7 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 →

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 →

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 →

Staff Machine Learning Engineer Remote

Lead the technical strategy and execution of ML initiatives for user growth and engagement in a fully remote team.

Phantom Remote Published 19 hours ago

Ray Sagemaker Snowflake Spark JAX

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 →