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

Showing 3 of 3 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 Bay Area, CA, United States of America Published 1 month ago

JAX Machine Learning Causal Reasoning representation learning reinforcement learning

Flexible on stack

Apply →

Applied Research Engineer, Agents Remote

Labelbox San Francisco Bay Area $250k–$300k/yr Published 11 months ago

diffusion RLAIF LLM agents JAX τ-bench

Apply →

Applied Research Scientist, Agents Remote

Labelbox San Francisco Bay Area $250k–$300k/yr Published 1 year ago

Multimodal Models Direct Preference Optimization (DPO) JAX Large Language Models Reinforcement Learning from Human Feedback (RLHF)

Apply →