jax

761bd42c - Merge pull request #27273 from carlosgmartin:fix_categorical_docs