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

Showing 3 of 3 results

BioMedical AI Research Engineer Hybrid

Join Xaira Therapeutics as a Biomedical AI Engineer to design AI systems that accelerate therapeutic discovery.

Xaira Therapeutics South San Francisco, California, United States $227k–$295k/yr Published 5 months ago

JAX Large Language Models biomedical datasets PyTorch TensorFlow

Flexible on stack

Apply →

AI Scientist - Biomedical Multimodal Modeling Hybrid

Join Xaira Therapeutics as an AI Scientist to develop multimodal foundation models for drug discovery using advanced AI techniques.

Xaira Therapeutics South San Francisco, California, United States $170k–$240k/yr Published 4 months ago

JAX multimodal learning generative modeling representation learning PyTorch

Flexible on stack

Apply →

Scientist, Machine Learning

Join Atomic AI as a Scientist to develop machine learning models for RNA structure prediction and drug targeting.

Atomic AI South San Francisco, California, United States $170k–$220k/yr Published 8 months ago

JAX Machine Learning PyTorch drug discovery Python

Flexible on stack

Apply →