"jax" Jobs

2 open tech roles matching “jax”, taken straight from company career pages — not reposted from other job boards. Most in demand right now: C++, Ray, TensorRT-LLM. Every listing is re-checked daily and closed roles are removed.

Showing 2 of 2 results

Computational Scientist, Differentiable Physics Visa

Join Periodic Labs as a Computational Scientist to develop differentiable simulations for continuum-physics problems using AI.

Periodic Labs Menlo Park, CA $250k–$350k/yr Published 1 week ago

C++ JAX fluid dynamics PyTorch continuum physics

Flexible on stack

Apply →

ML Systems Engineer Visa

Join Periodic Labs as an ML Systems Engineer to build and optimize large-scale training and reinforcement learning infrastructure.

Periodic Labs Menlo Park, CA $250k–$350k/yr Published 4 months ago

Ray TensorRT-LLM Triton SGLang CUDA

Flexible on stack

Apply →