Index

_

__call__() (transformer_engine.jax.flax.DenseGeneral method) (transformer_engine.jax.flax.DotProductAttention method) (transformer_engine.jax.flax.LayerNorm method) (transformer_engine.jax.flax.LayerNormDenseGeneral method) (transformer_engine.jax.flax.LayerNormMLP method) (transformer_engine.jax.flax.MultiHeadAttention method) (transformer_engine.jax.flax.RelativePositionBiases method) (transformer_engine.jax.flax.TransformerLayer method)

A

add() (transformer_engine.pytorch.CudaRNGStatesTracker method) analytics_anonymize_ip configuration value

analytics_id configuration value

B

bitbucket_url configuration value

C

canonical_url configuration value checkpoint() (in module transformer_engine.pytorch) collapse_navigation configuration value configuration value analytics_anonymize_ip analytics_id bitbucket_url canonical_url collapse_navigation display_version flyout_display github_url gitlab_url includehidden language_selector logo_only navigation_depth prev_next_buttons_location sticky_navigation style_external_links style_nav_header_background titles_only vcs_pageview_mode version_selector

cp_resource (transformer_engine.jax.MeshResource attribute) CudaRNGStatesTracker (class in transformer_engine.pytorch)

D

Data_item_1 (built-in variable) Data_item_2 (built-in variable) Data_item_3 (built-in variable) DelayedScaling (class in transformer_engine.common.recipe) DenseGeneral (class in transformer_engine.jax.flax) destroy_ub() (in module transformer_engine.pytorch) DisableFP8GEMM (class in transformer_engine.debug.features.disable_fp8_gemm)

DisableFP8Layer (class in transformer_engine.debug.features.disable_fp8_layer) display_version configuration value Documentation DotProductAttention (class in transformer_engine.jax.flax) (class in transformer_engine.pytorch) dp_resource (transformer_engine.jax.MeshResource attribute)

E

extend_logical_axis_rules() (in module transformer_engine.jax.flax)

F

FakeQuant (class in transformer_engine.debug.features.fake_quant) Float8BlockScaling (class in transformer_engine.common.recipe) Float8CurrentScaling (class in transformer_engine.common.recipe) flyout_display configuration value Foo (C++ function) fork() (transformer_engine.pytorch.CudaRNGStatesTracker method) Format (class in transformer_engine.common.recipe) forward() (transformer_engine.pytorch.DotProductAttention method) (transformer_engine.pytorch.GroupedLinear method) (transformer_engine.pytorch.LayerNormLinear method) (transformer_engine.pytorch.LayerNormMLP method) (transformer_engine.pytorch.Linear method) (transformer_engine.pytorch.MultiheadAttention method) (transformer_engine.pytorch.TransformerLayer method)

fp8_autocast() (in module transformer_engine.jax) (in module transformer_engine.pytorch) fp8_gemm_enabled() (in module transformer_engine.debug.features.api.TEDefaultFeatures) fp8_model_init() (in module transformer_engine.pytorch) fsdp_resource (transformer_engine.jax.MeshResource attribute)

G

get_cpu_offload_context() (in module transformer_engine.pytorch) get_states() (transformer_engine.pytorch.CudaRNGStatesTracker method) github_url configuration value

gitlab_url configuration value GroupedLinear (class in transformer_engine.pytorch)

I

includehidden configuration value initialize_ub() (in module transformer_engine.pytorch)

inspect_tensor() (in module transformer_engine.debug.features.api.TEDefaultFeatures) inspect_tensor_enabled() (in module transformer_engine.debug.features.api.TEDefaultFeatures) inspect_tensor_postquantize() (in module transformer_engine.debug.features.api.TEDefaultFeatures) inspect_tensor_postquantize_enabled() (in module transformer_engine.debug.features.api.TEDefaultFeatures)

L

language_selector configuration value LayerNorm (class in transformer_engine.jax.flax) (class in transformer_engine.pytorch) LayerNormDenseGeneral (class in transformer_engine.jax.flax) LayerNormLinear (class in transformer_engine.pytorch) LayerNormMLP (class in transformer_engine.jax.flax) (class in transformer_engine.pytorch)

Linear (class in transformer_engine.pytorch) List (C++ type) LogFp8TensorStats (class in transformer_engine.debug.features.log_fp8_tensor_stats) logo_only configuration value LogTensorStats (class in transformer_engine.debug.features.log_tensor_stats)

M

make_graphed_callables() (in module transformer_engine.pytorch) MeshResource (class in transformer_engine.jax) modify_tensor() (in module transformer_engine.debug.features.api.TEDefaultFeatures) modify_tensor_enabled() (in module transformer_engine.debug.features.api.TEDefaultFeatures) ModNested() (class) ModTopLevel() (class) module_a.submodule (module) module_b.submodule (module) moe_permute() (in module transformer_engine.pytorch) moe_permute_with_probs() (in module transformer_engine.pytorch) moe_sort_chunks_by_index() (in module transformer_engine.pytorch)

moe_sort_chunks_by_index_with_probs() (in module transformer_engine.pytorch) moe_unpermute() (in module transformer_engine.pytorch) MultiHeadAttention (class in transformer_engine.jax.flax) MultiheadAttention (class in transformer_engine.pytorch) MXFP8BlockScaling (class in transformer_engine.common.recipe) MyEnum (C++ enum) MyEnum::A (C++ enumerator) MyScopedEnum (C++ enum) MyScopedEnum::B (C++ enumerator) MyScopedVisibilityEnum (C++ enum) MyScopedVisibilityEnum::B (C++ enumerator) MyType (C++ type)

N

navigation_depth configuration value NVTE_Activation_Type (C++ enum) NVTE_Activation_Type::GEGLU (C++ enumerator) NVTE_Activation_Type::GELU (C++ enumerator) NVTE_Activation_Type::QGEGLU (C++ enumerator) NVTE_Activation_Type::QGELU (C++ enumerator) NVTE_Activation_Type::REGLU (C++ enumerator) NVTE_Activation_Type::RELU (C++ enumerator) NVTE_Activation_Type::SILU (C++ enumerator) NVTE_Activation_Type::SREGLU (C++ enumerator) NVTE_Activation_Type::SRELU (C++ enumerator) NVTE_Activation_Type::SWIGLU (C++ enumerator) NVTE_Bias_Type (C++ enum) NVTE_Bias_Type::NVTE_ALIBI (C++ enumerator) NVTE_Bias_Type::NVTE_NO_BIAS (C++ enumerator) NVTE_Bias_Type::NVTE_POST_SCALE_BIAS (C++ enumerator) NVTE_Bias_Type::NVTE_PRE_SCALE_BIAS (C++ enumerator) nvte_cast_transpose (C++ function) nvte_cast_transpose_dbias (C++ function) nvte_cast_transpose_dbias_dgelu (C++ function) nvte_cast_transpose_dbias_dqgelu (C++ function) nvte_cast_transpose_dbias_drelu (C++ function) nvte_cast_transpose_dbias_dsilu (C++ function) nvte_cast_transpose_dbias_dsrelu (C++ function) nvte_cast_transpose_with_noop (C++ function) nvte_compute_amax (C++ function) nvte_compute_scale_from_amax (C++ function) nvte_convert_bshd_to_thd (C++ function) nvte_convert_thd_to_bshd (C++ function) nvte_copy_to_kv_cache (C++ function) nvte_cp_thd_get_partitioned_indices (C++ function) nvte_cp_thd_grad_correction (C++ function) nvte_cp_thd_out_correction (C++ function) nvte_cp_thd_read_half_tensor (C++ function) nvte_cp_thd_read_second_half_lse (C++ function) nvte_cp_thd_second_half_lse_correction (C++ function) nvte_create_quantization_config (C++ function) nvte_create_tensor (C++ function) nvte_cublas_atomic_gemm (C++ function) nvte_cublas_gemm (C++ function) nvte_delayed_scaling_recipe_amax_and_scale_update (C++ function) nvte_delayed_scaling_recipe_amax_and_scale_update_after_reduction (C++ function) nvte_dequantize (C++ function) nvte_destroy_quantization_config (C++ function) nvte_destroy_tensor (C++ function) nvte_device_radix_sort_pairs (C++ function) nvte_dgeglu (C++ function) nvte_dgeglu_cast_transpose (C++ function) nvte_dgelu (C++ function) nvte_dqgeglu (C++ function) nvte_dqgeglu_cast_transpose (C++ function) nvte_dqgelu (C++ function) nvte_dreglu (C++ function) nvte_dreglu_cast_transpose (C++ function) nvte_drelu (C++ function) nvte_dsilu (C++ function) nvte_dsreglu (C++ function) nvte_dsreglu_cast_transpose (C++ function) nvte_dsrelu (C++ function) nvte_dswiglu (C++ function) nvte_dswiglu_cast_transpose (C++ function) nvte_enable_cudnn_norm_bwd (C++ function) nvte_enable_cudnn_norm_fwd (C++ function) nvte_enable_zero_centered_gamma_in_weight_dtype (C++ function) nvte_extract_seed_and_offset (C++ function) nvte_fp8_block_scaling_compute_partial_amax (C++ function) nvte_fp8_block_scaling_partial_cast (C++ function) nvte_fp8_transpose_dbias (C++ function) NVTE_Fused_Attn_Backend (C++ enum) NVTE_Fused_Attn_Backend::NVTE_F16_arbitrary_seqlen (C++ enumerator) NVTE_Fused_Attn_Backend::NVTE_F16_max512_seqlen (C++ enumerator) NVTE_Fused_Attn_Backend::NVTE_FP8 (C++ enumerator) NVTE_Fused_Attn_Backend::NVTE_No_Backend (C++ enumerator) nvte_fused_attn_bwd (C++ function) nvte_fused_attn_bwd_kvpacked (C++ function) nvte_fused_attn_bwd_qkvpacked (C++ function) nvte_fused_attn_fwd (C++ function) nvte_fused_attn_fwd_kvpacked (C++ function) nvte_fused_attn_fwd_qkvpacked (C++ function) nvte_fused_rope_backward (C++ function) nvte_fused_rope_forward (C++ function) nvte_geglu (C++ function) nvte_gelu (C++ function) nvte_get_fused_attn_backend (C++ function) nvte_get_kv_format (C++ function) nvte_get_q_format (C++ function) nvte_get_qkv_format (C++ function) nvte_get_qkv_layout_group (C++ function) nvte_get_quantization_config_attribute (C++ function) nvte_get_runtime_num_segments (C++ function) nvte_get_tensor_param (C++ function) nvte_is_non_tn_fp8_gemm_supported (C++ function) nvte_layernorm_bwd (C++ function) nvte_layernorm_fwd (C++ function) nvte_make_shape (C++ function) NVTE_Mask_Type (C++ enum) NVTE_Mask_Type::NVTE_CAUSAL_BOTTOM_RIGHT_MASK (C++ enumerator) NVTE_Mask_Type::NVTE_CAUSAL_MASK (C++ enumerator) NVTE_Mask_Type::NVTE_NO_MASK (C++ enumerator) NVTE_Mask_Type::NVTE_PADDING_CAUSAL_BOTTOM_RIGHT_MASK (C++ enumerator) NVTE_Mask_Type::NVTE_PADDING_CAUSAL_MASK (C++ enumerator) NVTE_Mask_Type::NVTE_PADDING_MASK (C++ enumerator) nvte_memset (C++ function) nvte_multi_cast_transpose (C++ function) nvte_multi_padding (C++ function) nvte_multi_stream_cublas_gemm (C++ function) nvte_multi_tensor_adam_capturable_cuda (C++ function) nvte_multi_tensor_adam_capturable_master_cuda (C++ function) nvte_multi_tensor_adam_cuda (C++ function) nvte_multi_tensor_adam_fp8_cuda (C++ function) nvte_multi_tensor_adam_param_remainder_cuda (C++ function) nvte_multi_tensor_compute_scale_and_scale_inv_cuda (C++ function) nvte_multi_tensor_l2norm_cuda (C++ function) nvte_multi_tensor_scale_cuda (C++ function) nvte_multi_tensor_sgd_cuda (C++ function) nvte_multi_tensor_unscale_l2norm_cuda (C++ function) NVTE_Norm_Type (C++ enum) NVTE_Norm_Type::LayerNorm (C++ enumerator) NVTE_Norm_Type::RMSNorm (C++ enumerator) nvte_permute (C++ function) nvte_populate_rng_state_async (C++ function) nvte_prepare_flash_attn_bwd (C++ function) nvte_prepare_flash_attn_fwd (C++ function) nvte_qgeglu (C++ function) nvte_qgelu (C++ function) NVTE_QKV_Format (C++ enum) NVTE_QKV_Format::NVTE_BSHD (C++ enumerator) NVTE_QKV_Format::NVTE_BSHD_2SBHD (C++ enumerator)

NVTE_QKV_Format::NVTE_SBHD (C++ enumerator) NVTE_QKV_Format::NVTE_SBHD_2BSHD (C++ enumerator) NVTE_QKV_Format::NVTE_THD (C++ enumerator) NVTE_QKV_Format::NVTE_THD_2BSHD (C++ enumerator) NVTE_QKV_Format::NVTE_THD_2SBHD (C++ enumerator) NVTE_QKV_Layout (C++ enum) NVTE_QKV_Layout::NVTE_BS3HD (C++ enumerator) NVTE_QKV_Layout::NVTE_BSH3D (C++ enumerator) NVTE_QKV_Layout::NVTE_BSHD_BS2HD (C++ enumerator) NVTE_QKV_Layout::NVTE_BSHD_BSH2D (C++ enumerator) NVTE_QKV_Layout::NVTE_BSHD_BSHD_BSHD (C++ enumerator) NVTE_QKV_Layout::NVTE_BSHD_SBHD_SBHD (C++ enumerator) NVTE_QKV_Layout::NVTE_Paged_KV_BSHD_BSHD_BSHD (C++ enumerator) NVTE_QKV_Layout::NVTE_Paged_KV_BSHD_SBHD_SBHD (C++ enumerator) NVTE_QKV_Layout::NVTE_Paged_KV_SBHD_BSHD_BSHD (C++ enumerator) NVTE_QKV_Layout::NVTE_Paged_KV_SBHD_SBHD_SBHD (C++ enumerator) NVTE_QKV_Layout::NVTE_Paged_KV_THD_BSHD_BSHD (C++ enumerator) NVTE_QKV_Layout::NVTE_Paged_KV_THD_SBHD_SBHD (C++ enumerator) NVTE_QKV_Layout::NVTE_SB3HD (C++ enumerator) NVTE_QKV_Layout::NVTE_SBH3D (C++ enumerator) NVTE_QKV_Layout::NVTE_SBHD_BSHD_BSHD (C++ enumerator) NVTE_QKV_Layout::NVTE_SBHD_SB2HD (C++ enumerator) NVTE_QKV_Layout::NVTE_SBHD_SBH2D (C++ enumerator) NVTE_QKV_Layout::NVTE_SBHD_SBHD_SBHD (C++ enumerator) NVTE_QKV_Layout::NVTE_T3HD (C++ enumerator) NVTE_QKV_Layout::NVTE_TH3D (C++ enumerator) NVTE_QKV_Layout::NVTE_THD_BSHD_BSHD (C++ enumerator) NVTE_QKV_Layout::NVTE_THD_SBHD_SBHD (C++ enumerator) NVTE_QKV_Layout::NVTE_THD_T2HD (C++ enumerator) NVTE_QKV_Layout::NVTE_THD_TH2D (C++ enumerator) NVTE_QKV_Layout::NVTE_THD_THD_THD (C++ enumerator) NVTE_QKV_Layout_Group (C++ enum) NVTE_QKV_Layout_Group::NVTE_3HD (C++ enumerator) NVTE_QKV_Layout_Group::NVTE_H3D (C++ enumerator) NVTE_QKV_Layout_Group::NVTE_HD_2HD (C++ enumerator) NVTE_QKV_Layout_Group::NVTE_HD_H2D (C++ enumerator) NVTE_QKV_Layout_Group::NVTE_HD_HD_HD (C++ enumerator) NVTE_QKV_Layout_Group::NVTE_Paged_KV_HD_HD_HD (C++ enumerator) nvte_quantize (C++ function) nvte_quantize_dbias (C++ function) nvte_quantize_dbias_dgelu (C++ function) nvte_quantize_dbias_dqgelu (C++ function) nvte_quantize_dbias_drelu (C++ function) nvte_quantize_dbias_dsilu (C++ function) nvte_quantize_dbias_dsrelu (C++ function) nvte_quantize_noop (C++ function) nvte_quantize_v2 (C++ function) nvte_reglu (C++ function) nvte_relu (C++ function) nvte_rmsnorm_bwd (C++ function) nvte_rmsnorm_fwd (C++ function) nvte_scaled_aligned_causal_masked_softmax_backward (C++ function) nvte_scaled_aligned_causal_masked_softmax_forward (C++ function) nvte_scaled_masked_softmax_backward (C++ function) nvte_scaled_masked_softmax_forward (C++ function) nvte_scaled_softmax_backward (C++ function) nvte_scaled_softmax_forward (C++ function) nvte_scaled_upper_triang_masked_softmax_backward (C++ function) nvte_scaled_upper_triang_masked_softmax_forward (C++ function) nvte_set_quantization_config_attribute (C++ function) nvte_set_tensor_param (C++ function) nvte_silu (C++ function) nvte_sreglu (C++ function) nvte_srelu (C++ function) nvte_swiglu (C++ function) nvte_swizzle_scaling_factors (C++ function) nvte_tensor_amax (C++ function) nvte_tensor_columnwise_data (C++ function) nvte_tensor_columnwise_shape (C++ function) nvte_tensor_data (C++ function) nvte_tensor_element_size (C++ function) nvte_tensor_ndims (C++ function) nvte_tensor_numel (C++ function) nvte_tensor_pack_create (C++ function) nvte_tensor_pack_destroy (C++ function) nvte_tensor_scale (C++ function) nvte_tensor_scale_inv (C++ function) nvte_tensor_scale_inv_shape (C++ function) nvte_tensor_scaling_mode (C++ function) nvte_tensor_shape (C++ function) nvte_tensor_size (C++ function) nvte_tensor_type (C++ function) nvte_transpose (C++ function) nvte_transpose_with_noop (C++ function) nvte_unpermute (C++ function) nvte_zero_tensor (C++ function) NVTEBasicTensor (C++ struct) NVTEBasicTensor::data_ptr (C++ member) NVTEBasicTensor::dtype (C++ member) NVTEBasicTensor::shape (C++ member) NVTEDType (C++ enum) NVTEDType::kNVTEBFloat16 (C++ enumerator) NVTEDType::kNVTEByte (C++ enumerator) NVTEDType::kNVTEFloat16 (C++ enumerator) NVTEDType::kNVTEFloat32 (C++ enumerator) NVTEDType::kNVTEFloat8E4M3 (C++ enumerator) NVTEDType::kNVTEFloat8E5M2 (C++ enumerator) NVTEDType::kNVTEFloat8E8M0 (C++ enumerator) NVTEDType::kNVTEInt16 (C++ enumerator) NVTEDType::kNVTEInt32 (C++ enumerator) NVTEDType::kNVTEInt64 (C++ enumerator) NVTEDType::kNVTENumTypes (C++ enumerator) NVTEQuantizationConfig (C++ type) NVTEQuantizationConfigAttribute (C++ enum) NVTEQuantizationConfigAttribute::kNVTEQuantizationConfigAmaxEpsilon (C++ enumerator) NVTEQuantizationConfigAttribute::kNVTEQuantizationConfigForcePow2Scales (C++ enumerator) NVTEQuantizationConfigAttribute::kNVTEQuantizationConfigNoopTensor (C++ enumerator) NVTEQuantizationConfigAttribute::kNVTEQuantizationConfigNumAttributes (C++ enumerator) NVTEScalingMode (C++ enum) NVTEScalingMode::NVTE_BLOCK_SCALING_1D (C++ enumerator) NVTEScalingMode::NVTE_BLOCK_SCALING_2D (C++ enumerator) NVTEScalingMode::NVTE_DELAYED_TENSOR_SCALING (C++ enumerator) NVTEScalingMode::NVTE_INVALID_SCALING (C++ enumerator) NVTEScalingMode::NVTE_MXFP8_1D_SCALING (C++ enumerator) NVTEShape (C++ struct) NVTEShape::data (C++ member) NVTEShape::ndim (C++ member) NVTETensor (C++ type) NVTETensorPack (C++ struct) NVTETensorPack::MAX_SIZE (C++ member) NVTETensorPack::size (C++ member) NVTETensorPack::tensors (C++ member) NVTETensorParam (C++ enum) NVTETensorParam::kNVTEAmax (C++ enumerator) NVTETensorParam::kNVTEColumnwiseData (C++ enumerator) NVTETensorParam::kNVTEColumnwiseScaleInv (C++ enumerator) NVTETensorParam::kNVTENumTensorParams (C++ enumerator) NVTETensorParam::kNVTERowwiseData (C++ enumerator) NVTETensorParam::kNVTERowwiseScaleInv (C++ enumerator) NVTETensorParam::kNVTEScale (C++ enumerator)

P

payments.dotpay.DotpayProvider (built-in class) PerTensorScaling (class in transformer_engine.debug.features.per_tensor_scaling) pp_resource (transformer_engine.jax.MeshResource attribute)

prev_next_buttons_location configuration value Python Enhancement Proposals PEP 287

R

Reading RelativePositionBiases (class in transformer_engine.jax.flax) reset() (transformer_engine.pytorch.CudaRNGStatesTracker method)

RFC RFC 2822 RMSNorm (class in transformer_engine.pytorch)

S

set_context_parallel_group() (transformer_engine.pytorch.DotProductAttention method) (transformer_engine.pytorch.MultiheadAttention method) (transformer_engine.pytorch.TransformerLayer method) set_states() (transformer_engine.pytorch.CudaRNGStatesTracker method) set_tensor_parallel_group() (transformer_engine.pytorch.GroupedLinear method) (transformer_engine.pytorch.LayerNormLinear method) (transformer_engine.pytorch.LayerNormMLP method) (transformer_engine.pytorch.Linear method) (transformer_engine.pytorch.MultiheadAttention method) (transformer_engine.pytorch.TransformerLayer method)

Sphinx::version (C++ member) std::array (C++ class) sticky_navigation configuration value style_external_links configuration value style_nav_header_background configuration value

T

titles_only configuration value tp_resource (transformer_engine.jax.MeshResource attribute) transformer_engine (C++ type), [1], [2] transformer_engine::DType (C++ enum) transformer_engine::DType::kBFloat16 (C++ enumerator) transformer_engine::DType::kByte (C++ enumerator) transformer_engine::DType::kFloat16 (C++ enumerator) transformer_engine::DType::kFloat32 (C++ enumerator) transformer_engine::DType::kFloat8E4M3 (C++ enumerator) transformer_engine::DType::kFloat8E5M2 (C++ enumerator) transformer_engine::DType::kFloat8E8M0 (C++ enumerator) transformer_engine::DType::kInt16 (C++ enumerator) transformer_engine::DType::kInt32 (C++ enumerator) transformer_engine::DType::kInt64 (C++ enumerator) transformer_engine::DType::kNumTypes (C++ enumerator) transformer_engine::is_fp8_dtype (C++ function) transformer_engine::num_streams (C++ member) transformer_engine::nvte_cublas_handle_init (C++ function) transformer_engine::nvte_cudnn_handle_init (C++ function) transformer_engine::QuantizationConfigWrapper (C++ struct) transformer_engine::QuantizationConfigWrapper::config_ (C++ member) transformer_engine::QuantizationConfigWrapper::operator NVTEQuantizationConfig (C++ function) transformer_engine::QuantizationConfigWrapper::operator= (C++ function), [1] transformer_engine::QuantizationConfigWrapper::QuantizationConfigWrapper (C++ function), [1], [2] transformer_engine::QuantizationConfigWrapper::set_amax_epsilon (C++ function) transformer_engine::QuantizationConfigWrapper::set_force_pow_2_scales (C++ function) transformer_engine::QuantizationConfigWrapper::set_noop_tensor (C++ function) transformer_engine::QuantizationConfigWrapper::~QuantizationConfigWrapper (C++ function) transformer_engine::TensorWrapper (C++ struct) transformer_engine::TensorWrapper::amax (C++ function) transformer_engine::TensorWrapper::bytes (C++ function) transformer_engine::TensorWrapper::columnwise_dptr (C++ function) transformer_engine::TensorWrapper::columnwise_shape (C++ function) transformer_engine::TensorWrapper::convertShape (C++ function), [1]

transformer_engine::TensorWrapper::data (C++ function) transformer_engine::TensorWrapper::defaultData (C++ member) transformer_engine::TensorWrapper::defaultShape (C++ member) transformer_engine::TensorWrapper::dptr (C++ function) transformer_engine::TensorWrapper::dtype (C++ function) transformer_engine::TensorWrapper::element_size (C++ function) transformer_engine::TensorWrapper::get_amax (C++ function) transformer_engine::TensorWrapper::get_columnwise_data (C++ function) transformer_engine::TensorWrapper::get_columnwise_scale_inv (C++ function) transformer_engine::TensorWrapper::get_parameter (C++ function) transformer_engine::TensorWrapper::get_rowwise_data (C++ function) transformer_engine::TensorWrapper::get_rowwise_scale_inv (C++ function) transformer_engine::TensorWrapper::get_scale (C++ function) transformer_engine::TensorWrapper::ndim (C++ function) transformer_engine::TensorWrapper::numel (C++ function) transformer_engine::TensorWrapper::operator= (C++ function), [1] transformer_engine::TensorWrapper::scale (C++ function) transformer_engine::TensorWrapper::scale_inv (C++ function) transformer_engine::TensorWrapper::scale_inv_shape (C++ function) transformer_engine::TensorWrapper::scaling_mode (C++ function) transformer_engine::TensorWrapper::set_amax (C++ function) transformer_engine::TensorWrapper::set_columnwise_data (C++ function) transformer_engine::TensorWrapper::set_columnwise_scale_inv (C++ function) transformer_engine::TensorWrapper::set_parameter (C++ function) transformer_engine::TensorWrapper::set_rowwise_data (C++ function) transformer_engine::TensorWrapper::set_rowwise_scale_inv (C++ function) transformer_engine::TensorWrapper::set_scale (C++ function) transformer_engine::TensorWrapper::shape (C++ function) transformer_engine::TensorWrapper::size (C++ function) transformer_engine::TensorWrapper::tensor_ (C++ member) transformer_engine::TensorWrapper::TensorWrapper (C++ function), [1], [2], [3], [4] transformer_engine::TensorWrapper::zero_ (C++ function) transformer_engine::TensorWrapper::~TensorWrapper (C++ function) TransformerLayer (class in transformer_engine.jax.flax) (class in transformer_engine.pytorch) TransformerLayerType (class in transformer_engine.jax.flax)

U

update_collections() (in module transformer_engine.jax)

V

vcs_pageview_mode configuration value

version (C++ member) version_selector configuration value

W

Writing