"jax" Jobs

5 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 5 of 5 results

Photonic IC Design Engineer, Design Automation & Modeling

Join K2 Space as a Photonic IC Design Engineer to lead photonic design automation and modeling for cutting-edge satellite technology.

K2 Space Seattle, WA $130k–$200k/yr Published 2 weeks ago

FEM JAX FDTD PyTorch Cadence

Flexible on stack

Apply →

Senior Software Engineer, ML Platform Hybrid

Join Xaira Therapeutics as a Senior Software Engineer to build AI infrastructure for drug discovery and development.

Xaira Therapeutics Seattle, Washington, United States $189k–$245k/yr Published 5 months ago

Torch Terraform JAX Ansible Python

Flexible on stack

Apply →

AI Research Engineer

Join a dynamic team as an AI Research Engineer to advance AI and machine learning in a startup focused on human-robot interaction.

Collaborative Robotics Seattle Published 1 year ago

JAX Diffusion Models reinforcement learning PyTorch TensorFlow

Flexible on stack

Apply →

Research Engineer / Research Scientist, RL Frontiers Hybrid Visa

Join Anthropic as a Research Engineer/Scientist to develop next-generation RL algorithms and architectures at frontier scale.

Anthropic San Francisco, CA | New York City, NY | Seattle, WA $500k–$850k/yr Published 1 month ago

JAX reinforcement learning PyTorch Python Transformer

Flexible on stack

Apply →

Research Engineer, RL Engineering Hybrid Visa

Anthropic San Francisco, CA | New York City, NY | Seattle, WA $500k–$850k/yr Published 1 year ago

distributed systems reinforcement learning LLM Python RLHF

Apply →