Closed kach closed 2 months ago
allow alice: chooses(a in A, to_maximize=f(a))
alice: chooses(a in A, to_maximize=f(a))
https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.argmax.html
https://jax.readthedocs.io/en/latest/_autosummary/jax.nn.one_hot.html
allow
alice: chooses(a in A, to_maximize=f(a))