juxai-s2 (pypi): JUX is a jax-accelerated engine for Lux-2022.
Keywords: JAX|LuxAI-S2