JAX jobs in California, USA

The goal is to build the first shared learned representation of our customers one dense vector per customer, trained on longitudinal transaction, sales and interaction history that downstream GenAI and analytics products can reuse instead of each re-deriving its own view of the same market. The contractor will design the pretraining objective, train and evaluate the encoder, and produce the evidence that determines whether the approach continues. Evaluation is as much of the deliverable as the