MANIFEST.in
setup.py
build_tools/VERSION.txt
build_tools/__init__.py
build_tools/build_ext.py
build_tools/jax.py
build_tools/paddle.py
build_tools/pytorch.py
build_tools/te_version.py
build_tools/utils.py
build_tools/__pycache__/__init__.cpython-310.pyc
build_tools/__pycache__/__init__.cpython-38.pyc
build_tools/__pycache__/build_ext.cpython-310.pyc
build_tools/__pycache__/build_ext.cpython-38.pyc
build_tools/__pycache__/jax.cpython-310.pyc
build_tools/__pycache__/te_version.cpython-310.pyc
build_tools/__pycache__/te_version.cpython-38.pyc
build_tools/__pycache__/utils.cpython-310.pyc
build_tools/__pycache__/utils.cpython-38.pyc
build_tools/wheel_utils/Dockerfile.aarch
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/cudnn_utils.h
common_headers/common/nvtx.h
common_headers/common/utils.cuh
common_headers/common/activation/activation_template.h
common_headers/common/comm_gemm_overlap/userbuffers/ipcsocket.h
common_headers/common/comm_gemm_overlap/userbuffers/userbuffers.h
common_headers/common/fused_attn/fused_attn_f16_arbitrary_seqlen.h
common_headers/common/fused_attn/fused_attn_f16_max512_seqlen.h
common_headers/common/fused_attn/fused_attn_fp8.h
common_headers/common/fused_attn/utils.h
common_headers/common/include/transformer_engine/activation.h
common_headers/common/include/transformer_engine/cast.h
common_headers/common/include/transformer_engine/cast_transpose_noop.h
common_headers/common/include/transformer_engine/comm_gemm_overlap.h
common_headers/common/include/transformer_engine/cudnn.h
common_headers/common/include/transformer_engine/fused_attn.h
common_headers/common/include/transformer_engine/fused_rope.h
common_headers/common/include/transformer_engine/gemm.h
common_headers/common/include/transformer_engine/layer_norm.h
common_headers/common/include/transformer_engine/padding.h
common_headers/common/include/transformer_engine/permutation.h
common_headers/common/include/transformer_engine/recipe.h
common_headers/common/include/transformer_engine/rmsnorm.h
common_headers/common/include/transformer_engine/softmax.h
common_headers/common/include/transformer_engine/transformer_engine.h
common_headers/common/include/transformer_engine/transpose.h
common_headers/common/layer_norm/ln.h
common_headers/common/layer_norm/ln_bwd_kernels.cuh
common_headers/common/layer_norm/ln_fwd_kernels.cuh
common_headers/common/layer_norm/ln_kernel_traits.h
common_headers/common/rmsnorm/rmsnorm.h
common_headers/common/rmsnorm/rmsnorm_bwd_kernels.cuh
common_headers/common/rmsnorm/rmsnorm_fwd_kernels.cuh
common_headers/common/rmsnorm/rmsnorm_kernel_traits.h
common_headers/common/util/cuda_driver.h
common_headers/common/util/cuda_runtime.h
common_headers/common/util/logging.h
common_headers/common/util/math.h
common_headers/common/util/pybind_helper.h
common_headers/common/util/rtc.h
common_headers/common/util/string.h
common_headers/common/util/system.h
common_headers/common/util/vectorized_pointwise.h
csrc/extensions.h
csrc/utils.cu
csrc/utils.h
csrc/extensions/activation.cpp
csrc/extensions/attention.cpp
csrc/extensions/cudnn.cpp
csrc/extensions/ffi.cpp
csrc/extensions/ffi.h
csrc/extensions/misc.cpp
csrc/extensions/misc.h
csrc/extensions/normalization.cpp
csrc/extensions/packing.cpp
csrc/extensions/pybind.cpp
csrc/extensions/quantization.cpp
csrc/extensions/softmax.cpp
csrc/extensions/transpose.cpp
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