>> Where does JAX shine vs. Julia? ...Documentation...Code Quality... I am not sure I need to go beyond that. The decision is somewhat easy here. I am surprise

>> Where does JAX shine vs. Julia? ...Documentation...Code Quality... I am not sure I need to go beyond that. The decision is somewhat easy here. I am surprised that integration is not mentioned. JAX is just Python (library) so it could be easily integrated into the rest of the system. It fits very well into Python/C skillset.