Skip to content
Open
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
fix
Signed-off-by: Pawel Gadzinski <[email protected]>
  • Loading branch information
pggPL committed Nov 4, 2025
commit 35e75de9705d13ad3e13d6636fda06ee57e277b2
2 changes: 1 addition & 1 deletion docs/api/jax.rst
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@ Jax
===

Pre-defined Variable of Logical Axes
-------------------------------------
------------------------------------
Variables are available in `transformer_engine.jax.sharding`.

* BATCH_AXES: The logical axis of batch dimension. It is usually sharded along DP + FSDP on Mesh.
Expand Down
Loading