Skip to content
Back to pull request #5102

Add out_sharding argument to call methods for layers with jax calls that support it #3397

Sign in to view logs

Add out_sharding argument to call methods for layers with jax calls that support it

Add out_sharding argument to call methods for layers with jax calls that support it #3397

Test pre-commit hooks

Test pre-commit hooks
succeeded Nov 19, 2025 in 19s