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

Showing 3 of 3 results

Senior Applied ML Engineer

Join CoreWeave as a Senior Applied ML Engineer to tackle challenges in continuous learning for AI agents with a focus on innovative solutions.

Coreweave Bellevue, WA / Sunnyvale, CA $182k–$242k/yr Published 5 months ago

Megatron JAX Temporal FastAPI PyTorch

Flexible on stack

Apply →

Senior Software Engineer - GPU Kernel Authoring & Optimization

Join CoreWeave as a Senior Engineer to optimize GPU kernels for high-performance AI applications in a rapidly growing environment.

Coreweave Sunnyvale, CA / Bellevue, WA $182k–$242k/yr Published 2 months ago

C++ JAX NVIDIA CUDA MLPerf

70% coding

Apply →

AI Engineer

Join Armada as an AI Engineer to build and deploy intelligent systems at the edge of the physical world.

Armada Bellevue Office, Sunset Corporate Campus $154.6k–$193.2k/yr Published 1 month ago

C++ JAX Java PyTorch TensorFlow

Flexible on stack 70% coding

Apply →