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

Showing 3 of 3 results

Research Engineer - Auto Research

Build agent-powered systems to improve model training and accelerate inference in AI drug design at Chai Discovery.

Chai Discovery San Francisco office Published 6 days ago

JAX AI/ML PyTorch Python

Flexible on stack

Apply →

Research Engineer - ML Infrastructure

Join Chai Discovery as a Research Engineer to optimize ML infrastructure for drug discovery in a fast-paced, innovative environment.

Chai Discovery San Francisco office Published 6 days ago

JAX PyTorch Python

Flexible on stack

Apply →

Research Engineer

Join Chai Discovery as an AI Research Engineer to advance AI drug design with a team of experts.

Chai Discovery San Francisco office Published 2 years ago

CUDA Slurm PyTorch Python Kubernetes

Flexible on stack

Apply →