Agent Executor (AX) is a Google research project for learning discrete choice models through differentiable maximum likelihood estimation. It is designed for situations where a model needs to predict choices from a finite set of alternatives, such as ranking, recommendation, preference modeling, or decision behavior analysis. The project provides JAX-based tools for defining and training choice models with automatic differentiation. It focuses on flexible model construction rather than a...