xma.utils

• xma.utils.contiguous
  • ensure_contiguous()
• xma.utils.cu_seqlens
• xma.utils.debugging
  • print_gradient()
• xma.utils.env
  • get_boolean_env_variable()
• xma.utils.packages
  • is_cute_dsl_available()
  • is_torch_neuronx_available()
  • is_torch_xla_available()
  • is_triton_available()
• xma.utils.ptx
  • get_ptx_from_triton_kernel()
• xma.utils.random
  • set_seed()
• xma.utils.settings
  • get_triton_num_warps()
• xma.utils.tensor
  • empty_like_contiguous()
  • get_alignment()
  • get_num_elements_and_hidden_size()
  • zeros_like_contiguous()