JAX Jobs (Remote)

6 open positions · Updated 6 days ago

6 roles Intern 2, Mid 1, Principal 1, Senior 1, Staff 1 2–8 yrs experience Bay Area, CA, United States of America +5 more Full-time, Internship newest 6 days ago

Also hiring: PyTorch Python Machine Learning TensorFlow Causal Reasoning Deep Learning

Showing 6 of 6 positions

Search with filters →

Principal Data Scientist, User Ecosystem Remote

Lead data science initiatives to enhance user engagement and ecosystem understanding at Reddit.

Reddit Remote - United States $313k–$365.1k/yr Published 1 week ago

JAX Machine Learning PyTorch Data Science

Flexible on stack

Apply →

Senior Machine Learning Engineer, Ads Response Prediction Remote

Join Instacart as a Senior Machine Learning Engineer to develop ML models for ads response prediction in a fully remote role.

Instacart Canada - Remote (ON, AB, BC, or NS Only) CA$180k–CA$190k/yr Published 1 month ago

pandas Spark 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 1 month 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 - Spain Published 4 months ago

SpaCy NLTK JAX PyTorch TensorFlow

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

JAX Machine Learning Causal Reasoning representation learning reinforcement learning

Flexible on stack

Apply →