"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

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

Torch Terraform JAX Ansible Python

Flexible on stack

Apply →

Staff Machine Learning Engineer (Modeling), Support

Lead the development of machine learning initiatives to enhance customer support experiences at Block.

Block Seattle, WA, United States of America $276.8k–$415.2k/yr Published 3 months ago

JAX PyTorch TensorFlow 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 10 months ago

JAX Diffusion Models reinforcement learning PyTorch TensorFlow

Flexible on stack

Apply →