MANIFEST.in
hipify_custom_map.json
pyproject.toml
setup.py
3rdparty/hipify_torch/.git
3rdparty/hipify_torch/.gitignore
3rdparty/hipify_torch/LICENSE.txt
3rdparty/hipify_torch/README.md
3rdparty/hipify_torch/hipify_cli.py
3rdparty/hipify_torch/setup.py
3rdparty/hipify_torch/cmake/Hipify.cmake
3rdparty/hipify_torch/hipify_torch/__init__.py
3rdparty/hipify_torch/hipify_torch/constants.py
3rdparty/hipify_torch/hipify_torch/cuda_to_hip_mappings.py
3rdparty/hipify_torch/hipify_torch/hipify_python.py
3rdparty/hipify_torch/hipify_torch/version.py
3rdparty/hipify_torch/hipify_torch/__pycache__/__init__.cpython-310.pyc
3rdparty/hipify_torch/hipify_torch/__pycache__/constants.cpython-310.pyc
3rdparty/hipify_torch/hipify_torch/__pycache__/cuda_to_hip_mappings.cpython-310.pyc
3rdparty/hipify_torch/hipify_torch/__pycache__/hipify_python.cpython-310.pyc
3rdparty/hipify_torch/hipify_torch/__pycache__/version.cpython-310.pyc
3rdparty/hipify_torch/hipify_torch/v2/__init__.py
3rdparty/hipify_torch/hipify_torch/v2/cuda_to_hip_mappings.py
3rdparty/hipify_torch/hipify_torch/v2/hipify_python.py
3rdparty/hipify_torch/test/test_installation.py
3rdparty/hipify_torch/tools/replace_cuda_with_hip_files.py
build_tools/VERSION.txt
build_tools/__init__.py
build_tools/build_ext.py
build_tools/jax.py
build_tools/pytorch.py
build_tools/te_version.py
build_tools/utils.py
build_tools/__pycache__/__init__.cpython-310.pyc
build_tools/__pycache__/build_ext.cpython-310.pyc
build_tools/__pycache__/jax.cpython-310.pyc
build_tools/__pycache__/te_version.cpython-310.pyc
build_tools/__pycache__/utils.cpython-310.pyc
build_tools/wheel_utils/Dockerfile.aarch
build_tools/wheel_utils/Dockerfile.rocm.manylinux.x86
build_tools/wheel_utils/Dockerfile.x86
build_tools/wheel_utils/build_wheels.sh
build_tools/wheel_utils/launch_aarch.sh
build_tools/wheel_utils/launch_x86.sh
common_headers/common/common.h
common_headers/common/common_hip.h
common_headers/common/cudnn_utils.h
common_headers/common/cudnn_utils_hip.h
common_headers/common/nvtx.h
common_headers/common/nvtx_hip.h
common_headers/common/utils.cuh
common_headers/common/utils_hip.cuh
common_headers/common/activation/activation_template.h
common_headers/common/activation/activation_template_hip.h
common_headers/common/amd_detail/hip_f8_impl.h
common_headers/common/amd_detail/hip_float8.h
common_headers/common/comm_gemm_overlap/userbuffers/ipcsocket.h
common_headers/common/comm_gemm_overlap/userbuffers/userbuffers.h
common_headers/common/comm_gemm_overlap/userbuffers/userbuffers_hip.h
common_headers/common/fused_attn/fused_attn_f16_arbitrary_seqlen.h
common_headers/common/fused_attn/fused_attn_f16_arbitrary_seqlen_hip.h
common_headers/common/fused_attn/fused_attn_f16_max512_seqlen.h
common_headers/common/fused_attn/fused_attn_f16_max512_seqlen_hip.h
common_headers/common/fused_attn/fused_attn_fp8.h
common_headers/common/fused_attn/fused_attn_fp8_hip.h
common_headers/common/fused_attn/utils.h
common_headers/common/fused_attn/utils_hip.h
common_headers/common/fused_attn_rocm/fused_attn_aotriton.h
common_headers/common/fused_attn_rocm/fused_attn_aotriton_hip.h
common_headers/common/fused_attn_rocm/fused_attn_ck.h
common_headers/common/fused_attn_rocm/fused_attn_ck_hip.h
common_headers/common/fused_attn_rocm/utils.h
common_headers/common/fused_attn_rocm/utils_hip.h
common_headers/common/fused_router/utils.h
common_headers/common/fused_router/utils_hip.h
common_headers/common/include/transformer_engine/activation.h
common_headers/common/include/transformer_engine/activation_hip.h
common_headers/common/include/transformer_engine/cast.h
common_headers/common/include/transformer_engine/cast_hip.h
common_headers/common/include/transformer_engine/cast_transpose_noop.h
common_headers/common/include/transformer_engine/cast_transpose_noop_hip.h
common_headers/common/include/transformer_engine/comm_gemm_overlap.h
common_headers/common/include/transformer_engine/comm_gemm_overlap_hip.h
common_headers/common/include/transformer_engine/cudnn.h
common_headers/common/include/transformer_engine/cudnn_hip.h
common_headers/common/include/transformer_engine/fused_attn.h
common_headers/common/include/transformer_engine/fused_attn_hip.h
common_headers/common/include/transformer_engine/fused_rope.h
common_headers/common/include/transformer_engine/fused_rope_hip.h
common_headers/common/include/transformer_engine/fused_router.h
common_headers/common/include/transformer_engine/fused_router_hip.h
common_headers/common/include/transformer_engine/gemm.h
common_headers/common/include/transformer_engine/gemm_hip.h
common_headers/common/include/transformer_engine/multi_stream.h
common_headers/common/include/transformer_engine/multi_stream_hip.h
common_headers/common/include/transformer_engine/multi_tensor.h
common_headers/common/include/transformer_engine/multi_tensor_hip.h
common_headers/common/include/transformer_engine/normalization.h
common_headers/common/include/transformer_engine/normalization_hip.h
common_headers/common/include/transformer_engine/padding.h
common_headers/common/include/transformer_engine/padding_hip.h
common_headers/common/include/transformer_engine/permutation.h
common_headers/common/include/transformer_engine/permutation_hip.h
common_headers/common/include/transformer_engine/recipe.h
common_headers/common/include/transformer_engine/recipe_hip.h
common_headers/common/include/transformer_engine/softmax.h
common_headers/common/include/transformer_engine/softmax_hip.h
common_headers/common/include/transformer_engine/swizzle.h
common_headers/common/include/transformer_engine/swizzle_hip.h
common_headers/common/include/transformer_engine/transformer_engine.h
common_headers/common/include/transformer_engine/transformer_engine_hip.h
common_headers/common/include/transformer_engine/transpose.h
common_headers/common/include/transformer_engine/transpose_hip.h
common_headers/common/multi_tensor/multi_tensor_apply.cuh
common_headers/common/multi_tensor/multi_tensor_apply_hip.cuh
common_headers/common/normalization/common.h
common_headers/common/normalization/common_hip.h
common_headers/common/normalization/kernel_traits.h
common_headers/common/normalization/kernel_traits_hip.h
common_headers/common/normalization/layernorm/ln_bwd_kernels.cuh
common_headers/common/normalization/layernorm/ln_bwd_kernels_hip.cuh
common_headers/common/normalization/layernorm/ln_fwd_kernels.cuh
common_headers/common/normalization/layernorm/ln_fwd_kernels_hip.cuh
common_headers/common/normalization/rmsnorm/rmsnorm_bwd_kernels.cuh
common_headers/common/normalization/rmsnorm/rmsnorm_bwd_kernels_hip.cuh
common_headers/common/normalization/rmsnorm/rmsnorm_fwd_kernels.cuh
common_headers/common/normalization/rmsnorm/rmsnorm_fwd_kernels_hip.cuh
common_headers/common/nvshmem_api/nvshmem_waitkernel.h
common_headers/common/nvshmem_api/nvshmem_waitkernel_hip.h
common_headers/common/recipe/recipe_common.cuh
common_headers/common/recipe/recipe_common_hip.cuh
common_headers/common/transpose/cast_transpose.h
common_headers/common/transpose/cast_transpose_hip.h
common_headers/common/util/cast_gated_kernels.cuh
common_headers/common/util/cast_gated_kernels_hip.cuh
common_headers/common/util/cast_kernels.cuh
common_headers/common/util/cast_kernels_hip.cuh
common_headers/common/util/cuda_driver.h
common_headers/common/util/cuda_nvml.h
common_headers/common/util/cuda_runtime.h
common_headers/common/util/dequantize_kernels.cuh
common_headers/common/util/dequantize_kernels_hip.cuh
common_headers/common/util/handle_manager.h
common_headers/common/util/handle_manager_hip.h
common_headers/common/util/hip_driver.h
common_headers/common/util/hip_nvml.h
common_headers/common/util/hip_runtime.h
common_headers/common/util/logging.h
common_headers/common/util/logging_hip.h
common_headers/common/util/math.h
common_headers/common/util/multi_stream.h
common_headers/common/util/multi_stream_hip.h
common_headers/common/util/ptx.cuh
common_headers/common/util/ptx_hip.cuh
common_headers/common/util/pybind_helper.h
common_headers/common/util/pybind_helper_hip.h
common_headers/common/util/rocm_cast_gated_kernels.cuh
common_headers/common/util/rocm_cast_gated_kernels_hip.cuh
common_headers/common/util/rocm_cast_kernels.cuh
common_headers/common/util/rocm_cast_kernels_hip.cuh
common_headers/common/util/rocm_dequantize_kernels.cuh
common_headers/common/util/rocm_dequantize_kernels_hip.cuh
common_headers/common/util/rocm_vectorized_2d.cuh
common_headers/common/util/rocm_vectorized_2d_hip.cuh
common_headers/common/util/rtc.h
common_headers/common/util/rtc_hip.h
common_headers/common/util/shared_lib_wrapper.h
common_headers/common/util/string.h
common_headers/common/util/system.h
common_headers/common/util/system_hip.h
common_headers/common/util/vectorized_pointwise.h
common_headers/common/util/vectorized_pointwise_hip.h
csrc/extensions.h
csrc/extensions/activation.cpp
csrc/extensions/attention.cpp
csrc/extensions/cublas.cpp
csrc/extensions/cudnn.cpp
csrc/extensions/ffi.cpp
csrc/extensions/ffi.h
csrc/extensions/gemm.cpp
csrc/extensions/misc.cpp
csrc/extensions/misc.h
csrc/extensions/normalization.cpp
csrc/extensions/pybind.cpp
csrc/extensions/quantization.cpp
csrc/extensions/softmax.cpp
csrc/extensions/utils.cpp
csrc/extensions/utils.h
transformer_engine_jax.egg-info/PKG-INFO
transformer_engine_jax.egg-info/SOURCES.txt
transformer_engine_jax.egg-info/dependency_links.txt
transformer_engine_jax.egg-info/requires.txt
transformer_engine_jax.egg-info/top_level.txt