cuequivariance-ops-cu12 0.4.0__py3-none-manylinux_2_24_x86_64.manylinux_2_28_x86_64.whl

cuequivariance_ops/VERSION

@@ -0,0 +1 @@
+0.4.0

cuequivariance_ops/__init__.py

@@ -0,0 +1,41 @@
+# SPDX-FileCopyrightText: Copyright (c) 2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: LicenseRef-NvidiaProprietary
+#
+# NVIDIA CORPORATION, its affiliates and licensors retain all intellectual
+# property and proprietary rights in and to this material, related
+# documentation and any modifications thereto. Any use, reproduction,
+# disclosure or distribution of this material and related documentation
+# without an express license agreement from NVIDIA CORPORATION or
+# its affiliates is strictly prohibited.
+
+from ._version import __version__, __git_commit__
+import os
+import ctypes
+
+PREFERRED_LOAD_FLAG = ctypes.RTLD_LOCAL
+
+
+def root_dir():
+    try:
+        import importlib.metadata
+
+        dist = importlib.metadata.distribution("cuequivariance_ops")
+        root = dist.locate_file("cuequivariance_ops")
+    except Exception:
+        # last resort, will fail with writeable install
+        root = os.path.dirname(__file__)
+    return root
+
+
+def load_library():
+    try:
+        ctypes.CDLL(
+            os.path.join(root_dir(), "lib/libcue_ops.so"), mode=PREFERRED_LOAD_FLAG
+        )
+    except Exception:
+        pass
+
+
+load_library()
+
+__all__ = ["__version__", "__git_commit__", "root_dir", "load_library"]

cuequivariance_ops/_version.py

@@ -0,0 +1,20 @@
+# SPDX-FileCopyrightText: Copyright (c) 2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: LicenseRef-NvidiaProprietary
+#
+# NVIDIA CORPORATION, its affiliates and licensors retain all intellectual
+# property and proprietary rights in and to this material, related
+# documentation and any modifications thereto. Any use, reproduction,
+# disclosure or distribution of this material and related documentation
+# without an express license agreement from NVIDIA CORPORATION or
+# its affiliates is strictly prohibited.
+
+
+import importlib.resources
+
+__version__ = (
+    importlib.resources.files("cuequivariance_ops")
+    .joinpath("VERSION")
+    .read_text()
+    .strip()
+)
+__git_commit__ = "release"

cuequivariance_ops/common/common.hpp

@@ -0,0 +1,98 @@
+/*
+ * Copyright (c) 2024, NVIDIA CORPORATION. All rights reserved.
+ *
+ * This source code and/or documentation ("Licensed Deliverables") are
+ * subject to NVIDIA intellectual property rights under U.S. and
+ * international Copyright laws.
+ */
+
+#pragma once
+
+#include <cstdint>
+#include <cuda_bf16.h>
+#include <cuda_fp16.h>
+#include <vector>
+
+namespace kernelcatcher {
+
+namespace tensor_product {
+/**
+ * @brief a wrapper struct containing informations
+ * about tensor-product paths
+ */
+template <typename MathT>
+struct __attribute__((aligned(16))) tp_info {
+  /** offsets into `path_offsets_and_dims` for each "target" */
+  const int32_t* __restrict__ path_csr_offsets{nullptr};
+  /** "sources" of all paths and their offsets and dimensions */
+  const int32_t* __restrict__ path_offsets_and_dims{nullptr};
+  /** clebsch-gordan values for each path */
+  const MathT* __restrict__ path_cg_values{nullptr};
+  /** number of "target" segments */
+  int32_t num_target_segments{0};
+  /** number of path (i.e. all paths between segments) */
+  int32_t num_paths{0};
+};  // struct tp_info
+
+enum class ConnectionModeT : uint8_t {
+  kUVW = 0,
+  // UVW with U spherical harmonic
+  k1VW,  // NOLINT
+  // UVW with V spherical harmonic
+  kU1W,  // NOLINT
+  kUVU,
+  kUVV,
+  kUUW,
+  kUUU,
+  // FullTP, no weight
+  kUVUV,
+  // FullTP, U spherical harmonic
+  k1V1V,
+  // FullTP, V spherical harmonic
+  kU1U1,
+  // Linear
+  kUUVV,
+};
+}  // namespace tensor_product
+
+namespace symmetric_tensor_contraction {
+/**
+ * @brief a wrapper struct containing informations
+ * about tensor-product paths
+ */
+template <typename DataT>
+struct __attribute__((aligned(16))) clebsch_gordan_tensor {
+  const DataT* __restrict__ cg_values{nullptr};
+  const int16_t* __restrict__ cg_indices{nullptr};
+  const int32_t* __restrict__ cg_offsets{nullptr};
+  int32_t total_output_irreps{0};
+};  // struct clebsch_gordan_tensor
+}  // namespace symmetric_tensor_contraction
+
+namespace batch_linear {
+struct __attribute__((aligned(8))) MatrixLayout {
+  int32_t size_row;  // uncontracted mode
+  int32_t size_col;  // contracted mode
+};
+
+struct __attribute__((aligned(8))) IndexOffset {
+  int32_t start;
+  int32_t end;
+};
+
+enum class GemvModeT : std::uint8_t { kUVV = 0, kUUV = 1 };
+enum class WeightSharedModeT : std::int32_t { kShared = 0, kIndexed = 1, kBatched = 2 };
+/**
+ * @brief a wrapper struct containing informations
+ * about tensor-product paths
+ */
+template <typename DataT>
+struct __attribute__((aligned(16))) batch_linear_info {
+  const MatrixLayout* __restrict__ layouts{nullptr};
+  const IndexOffset* __restrict__ index_offsets{nullptr};
+  const int32_t* __restrict__ indices{nullptr};
+  const DataT* __restrict__ alpha{nullptr};
+};  // struct batch_linear_info
+
+}  // namespace batch_linear
+}  // namespace kernelcatcher

cuequivariance_ops/common/nvtx.hpp

@@ -0,0 +1,29 @@
+/*
+ * Copyright (c) 2019-2023, NVIDIA CORPORATION. All rights reserved.
+ *
+ * This source code and/or documentation ("Licensed Deliverables") are
+ * subject to NVIDIA intellectual property rights under U.S. and
+ * international Copyright laws.
+ */
+
+#pragma once
+
+namespace kernelcatcher::utils {
+
+/**
+ * @brief Push a named nvtx range
+ * @param name range name
+ */
+void push_range(const char* name);
+
+/** Pop the latest range */
+void pop_range();
+
+struct range_guard {
+  range_guard(const char* name) { push_range(name); }
+  ~range_guard() { pop_range(); }
+  range_guard(range_guard const&)            = delete;
+  range_guard& operator=(range_guard const&) = delete;
+};
+
+}  // namespace kernelcatcher::utils

cuequivariance_ops/equivariance/fused_tensor_product.cuh

@@ -0,0 +1,297 @@
+/*
+ * Copyright (c) 2021-2024, NVIDIA CORPORATION. All rights reserved.
+ *
+ * This source code and/or documentation ("Licensed Deliverables") are
+ * subject to NVIDIA intellectual property rights under U.S. and
+ * international Copyright laws.
+ */
+
+#pragma once
+
+#include "../common/common.hpp"
+
+#include <algorithm>
+#include <limits>
+
+namespace kernelcatcher::tensor_product {
+
+struct __attribute__((aligned(16))) tp_data_sizes {
+  int64_t batch_size;
+  bool shared_a;
+  bool shared_b;
+  bool shared_w;
+  int32_t stride_a;
+  int32_t stride_b;
+  int32_t stride_w;
+  int32_t stride_o;
+};  // struct tp_data_sizes
+
+template <typename DataAT, typename DataBT, typename DataWeightT, typename DataOutT, typename MathT>
+void fused_tensor_product_fwd(DataOutT* out,
+                              const DataAT* in_a,
+                              const DataBT* in_b,
+                              const DataWeightT* weight,
+                              ConnectionModeT mode,
+                              const tp_info<MathT>& info,
+                              const tp_data_sizes& sizes,
+                              cudaStream_t stream);
+
+template <typename DataAT, typename DataBT, typename DataWeightT, typename DataOutT, typename MathT>
+void fused_tensor_product_bwd(DataAT* grad_in_a,
+                              DataBT* grad_in_b,
+                              DataWeightT* grad_weight,
+                              const DataOutT* grad_out,
+                              const DataAT* in_a,
+                              const DataBT* in_b,
+                              const DataWeightT* weight,
+                              ConnectionModeT mode,
+                              const tp_info<MathT>& info_bwd_dgrad_a,
+                              const tp_info<MathT>& info_bwd_dgrad_b,
+                              const tp_info<MathT>& info_bwd_dgrad_w,
+                              const tp_data_sizes& sizes,
+                              cudaStream_t stream);
+
+template <typename DataAT, typename DataBT, typename DataWeightT, typename DataOutT, typename MathT>
+void fused_tensor_product_bwd_bwd(DataAT* grad_in_a,
+                                  DataBT* grad_in_b,
+                                  DataWeightT* grad_weight,
+                                  DataOutT* grad_grad_out,
+                                  const DataAT* grad_grad_in_a,
+                                  const DataBT* grad_grad_in_b,
+                                  const DataWeightT* grad_grad_weight,
+                                  const DataOutT* grad_out,
+                                  const DataAT* in_a,
+                                  const DataBT* in_b,
+                                  const DataWeightT* weight,
+                                  ConnectionModeT mode,
+                                  const tp_info<MathT>& info_fwd,
+                                  const tp_info<MathT>& info_bwd_dgrad_a,
+                                  const tp_info<MathT>& info_bwd_dgrad_b,
+                                  const tp_info<MathT>& info_bwd_dgrad_w,
+                                  const tp_data_sizes& sizes,
+                                  cudaStream_t stream);
+
+extern template void fused_tensor_product_bwd_bwd<float, float, float, float, float>(
+  float*,
+  float*,
+  float*,
+  float*,
+  const float*,
+  const float*,
+  const float*,
+  const float*,
+  const float*,
+  const float*,
+  const float*,
+  ConnectionModeT,
+  const tp_info<float>&,
+  const tp_info<float>&,
+  const tp_info<float>&,
+  const tp_info<float>&,
+  const tp_data_sizes&,
+  cudaStream_t);
+
+extern template void fused_tensor_product_bwd_bwd<float, float, float, float, double>(
+  float*,
+  float*,
+  float*,
+  float*,
+  const float*,
+  const float*,
+  const float*,
+  const float*,
+  const float*,
+  const float*,
+  const float*,
+  ConnectionModeT,
+  const tp_info<double>&,
+  const tp_info<double>&,
+  const tp_info<double>&,
+  const tp_info<double>&,
+  const tp_data_sizes&,
+  cudaStream_t);
+
+extern template void fused_tensor_product_bwd_bwd<double, double, double, double, double>(
+  double*,
+  double*,
+  double*,
+  double*,
+  const double*,
+  const double*,
+  const double*,
+  const double*,
+  const double*,
+  const double*,
+  const double*,
+  ConnectionModeT,
+  const tp_info<double>&,
+  const tp_info<double>&,
+  const tp_info<double>&,
+  const tp_info<double>&,
+  const tp_data_sizes&,
+  cudaStream_t);
+
+extern template void fused_tensor_product_bwd_bwd<__half, __half, __half, __half, float>(
+  __half*,
+  __half*,
+  __half*,
+  __half*,
+  const __half*,
+  const __half*,
+  const __half*,
+  const __half*,
+  const __half*,
+  const __half*,
+  const __half*,
+  ConnectionModeT,
+  const tp_info<float>&,
+  const tp_info<float>&,
+  const tp_info<float>&,
+  const tp_info<float>&,
+  const tp_data_sizes&,
+  cudaStream_t);
+extern template void
+fused_tensor_product_bwd_bwd<__nv_bfloat16, __nv_bfloat16, __nv_bfloat16, __nv_bfloat16, float>(
+  __nv_bfloat16*,
+  __nv_bfloat16*,
+  __nv_bfloat16*,
+  __nv_bfloat16*,
+  const __nv_bfloat16*,
+  const __nv_bfloat16*,
+  const __nv_bfloat16*,
+  const __nv_bfloat16*,
+  const __nv_bfloat16*,
+  const __nv_bfloat16*,
+  const __nv_bfloat16*,
+  ConnectionModeT,
+  const tp_info<float>&,
+  const tp_info<float>&,
+  const tp_info<float>&,
+  const tp_info<float>&,
+  const tp_data_sizes&,
+  cudaStream_t);
+
+extern template void fused_tensor_product_bwd<float, float, float, float, float>(
+  float*,
+  float*,
+  float*,
+  const float*,
+  const float*,
+  const float*,
+  const float*,
+  ConnectionModeT,
+  const tp_info<float>&,
+  const tp_info<float>&,
+  const tp_info<float>&,
+  const tp_data_sizes&,
+  cudaStream_t);
+
+extern template void fused_tensor_product_bwd<float, float, float, float, double>(
+  float*,
+  float*,
+  float*,
+  const float*,
+  const float*,
+  const float*,
+  const float*,
+  ConnectionModeT,
+  const tp_info<double>&,
+  const tp_info<double>&,
+  const tp_info<double>&,
+  const tp_data_sizes&,
+  cudaStream_t);
+
+extern template void fused_tensor_product_bwd<double, double, double, double, double>(
+  double*,
+  double*,
+  double*,
+  const double*,
+  const double*,
+  const double*,
+  const double*,
+  ConnectionModeT,
+  const tp_info<double>&,
+  const tp_info<double>&,
+  const tp_info<double>&,
+  const tp_data_sizes&,
+  cudaStream_t);
+
+extern template void fused_tensor_product_bwd<__half, __half, __half, __half, float>(
+  __half*,
+  __half*,
+  __half*,
+  const __half*,
+  const __half*,
+  const __half*,
+  const __half*,
+  ConnectionModeT,
+  const tp_info<float>&,
+  const tp_info<float>&,
+  const tp_info<float>&,
+  const tp_data_sizes&,
+  cudaStream_t);
+extern template void
+fused_tensor_product_bwd<__nv_bfloat16, __nv_bfloat16, __nv_bfloat16, __nv_bfloat16, float>(
+  __nv_bfloat16*,
+  __nv_bfloat16*,
+  __nv_bfloat16*,
+  const __nv_bfloat16*,
+  const __nv_bfloat16*,
+  const __nv_bfloat16*,
+  const __nv_bfloat16*,
+  ConnectionModeT,
+  const tp_info<float>&,
+  const tp_info<float>&,
+  const tp_info<float>&,
+  const tp_data_sizes&,
+  cudaStream_t);
+
+extern template void fused_tensor_product_fwd<float, float, float, float, float>(
+  float*,
+  const float*,
+  const float*,
+  const float*,
+  ConnectionModeT,
+  const tp_info<float>&,
+  const tp_data_sizes&,
+  cudaStream_t);
+extern template void fused_tensor_product_fwd<float, float, float, float, double>(
+  float*,
+  const float*,
+  const float*,
+  const float*,
+  ConnectionModeT,
+  const tp_info<double>&,
+  const tp_data_sizes&,
+  cudaStream_t);
+extern template void fused_tensor_product_fwd<double, double, double, double, double>(
+  double*,
+  const double*,
+  const double*,
+  const double*,
+  ConnectionModeT,
+  const tp_info<double>&,
+  const tp_data_sizes&,
+  cudaStream_t);
+
+extern template void fused_tensor_product_fwd<__half, __half, __half, __half, float>(
+  __half*,
+  const __half*,
+  const __half*,
+  const __half*,
+  ConnectionModeT,
+  const tp_info<float>&,
+  const tp_data_sizes&,
+  cudaStream_t);
+extern template void
+fused_tensor_product_fwd<__nv_bfloat16, __nv_bfloat16, __nv_bfloat16, __nv_bfloat16, float>(
+  __nv_bfloat16*,
+  const __nv_bfloat16*,
+  const __nv_bfloat16*,
+  const __nv_bfloat16*,
+  ConnectionModeT,
+  const tp_info<float>&,
+  const tp_data_sizes&,
+  cudaStream_t);
+
+}  // namespace kernelcatcher::tensor_product

cuequivariance_ops/equivariance/segmented_transpose.cuh

@@ -0,0 +1,40 @@
+/*
+ * Copyright (c) 2021-2024, NVIDIA CORPORATION. All rights reserved.
+ *
+ * This source code and/or documentation ("Licensed Deliverables") are
+ * subject to NVIDIA intellectual property rights under U.S. and
+ * international Copyright laws.
+ */
+
+#pragma once
+
+#include "../common/common.hpp"
+
+namespace kernelcatcher::tensor_product {
+
+template <typename DataT>
+void segmented_transpose(DataT* tensor_transpose,
+                         const DataT* tensor,
+                         const int32_t* segment_info,
+                         int32_t num_segments,
+                         int64_t batch_size,
+                         int64_t stride,
+                         bool input_contiguous_as_info,
+                         cudaStream_t stream);
+
+extern template void segmented_transpose<float>(
+  float*, const float*, const int32_t*, int32_t, int64_t, int64_t, bool, cudaStream_t);
+extern template void segmented_transpose<double>(
+  double*, const double*, const int32_t*, int32_t, int64_t, int64_t, bool, cudaStream_t);
+extern template void segmented_transpose<__nv_bfloat16>(__nv_bfloat16*,
+                                                        const __nv_bfloat16*,
+                                                        const int32_t*,
+                                                        int32_t,
+                                                        int64_t,
+                                                        int64_t,
+                                                        bool,
+                                                        cudaStream_t);
+extern template void segmented_transpose<__half>(
+  __half*, const __half*, const int32_t*, int32_t, int64_t, int64_t, bool, cudaStream_t);
+
+}  // namespace kernelcatcher::tensor_product

cuequivariance_ops/equivariance/tensor_product_uniform_1d_jit.hh

@@ -0,0 +1,56 @@
+/*
+ * Copyright (c) 2025, NVIDIA CORPORATION. All rights reserved.
+ *
+ * This source code and/or documentation ("Licensed Deliverables") are
+ * subject to NVIDIA intellectual property rights under U.S. and
+ * international Copyright laws.
+ */
+
+#pragma once
+
+#include <cstdint>
+#include <string>
+#include <vector>
+
+namespace kernelcatcher::equivariance::tensor_product_uniform_1d_jit {
+
+enum class Datatype : int {
+  kFloat32  = 0,
+  kFloat64  = 1,
+  kFloat16  = 2,
+  kBFloat16 = 3,
+  kInt32    = 4,
+  kInt64    = 5
+};
+enum class Dimension : int { kScalar = 0, kOneDimensional = 1 };
+enum class BatchDimension : int { kBatched = 0, kShared = 1, kIndexed = 2 };
+
+extern int run_tensor_product_uniform_1d_jit(
+  std::string const& name,
+  Datatype math_dtype,
+  int operand_extent,
+  int num_inputs,
+  int num_outputs,
+  int num_index,
+  std::vector<Dimension> const& buffer_dim,     // num_inputs + num_outputs
+  std::vector<int> const& buffer_num_segments,  // num_inputs + num_outputs
+  std::vector<std::vector<BatchDimension>> const&
+    batch_dim,  // (num_inputs + num_outputs + num_index) x num_batch_axes
+  std::vector<std::vector<int>> const&
+    index_buffer,  // (num_inputs + num_outputs + num_index) x num_batch_axes
+  // ^^^^^ ignored unless batch_dim[i][j] is indexed
+  std::vector<int> const& index_extent,  // num_index
+  std::vector<Datatype> const& dtypes,   // num_inputs + num_outputs + num_index
+  std::vector<std::vector<int>> const& operations,
+  std::vector<int> num_paths,                // num_operations
+  std::vector<int> path_indices_start,       // num_operations
+  std::vector<int> path_coefficients_start,  // num_operations
+  std::vector<int> const& path_indices,
+  std::vector<double> const& path_coefficients,
+  std::vector<int> const& batch_sizes,  // num_batch_axes
+  std::vector<void*> const& buffers,
+  std::vector<size_t> const& buffer_bytes,
+  bool zero_output_buffers,
+  void* stream);
+
+}  // namespace kernelcatcher::equivariance::tensor_product_uniform_1d_jit

cuequivariance_ops_cu12-0.4.0.dist-info/METADATA

@@ -0,0 +1,203 @@
+Metadata-Version: 2.1
+Name: cuequivariance-ops-cu12
+Version: 0.4.0
+Summary: cuequivariance-ops - GPU Accelerated Extensions for Equivariant Primitives
+Author: NVIDIA Corporation
+License: # Software License Agreement
+        
+        LICENSE AGREEMENT FOR NVIDIA MATH LIBRARIES SOFTWARE DEVELOPMENT KITS
+        
+        This license agreement(“Agreement”) is a legal agreement between you and NVIDIA Corporation (“NVIDIA”) and governs your use of the NVIDIA math libraries software development kit as available at NVIDIA’s discretion (each, a “SDK”).
+        
+        Each SDK has its own set of software and materials, but here is a description of the types of items that may be included in a SDK: source code, header files, APIs, data sets and assets (examples include images, textures, models, scenes, videos, native API input/output files), binary software, sample code, libraries, utility programs, programming code and documentation.
+        
+        This Agreement can be accepted only by an adult of legal age of majority in the country in which the SDK is used.
+        
+        If you are entering into this Agreement on behalf of a company or other legal entity, you represent that you have the legal authority to bind the entity to this Agreement, in which case “you” will mean the entity you represent.
+        
+        If you don’t have the required age or authority to accept this Agreement, or if you don’t accept all the terms and conditions of this Agreement, do not download, install or use the SDK.
+        
+        You agree to use the SDK only for purposes that are permitted by (a) this Agreement, and (b) any applicable law, regulation or generally accepted practices or guidelines in the relevant jurisdictions.
+        
+            License.
+        
+        1.1 Grant
+        
+        Subject to the terms of this Agreement, NVIDIA hereby grants you a non-exclusive, non-transferable license, without the right to sublicense (except as expressly provided in this Agreement) to:
+        
+            Install and use the SDK, and
+        
+            Distribute the binary files, files identified as samples, and headers as incorporated into a software application that meets the distribution requirements indicated in this Agreement.
+        
+        1.2 Distribution Requirements
+        
+        These are the distribution requirements for you to exercise the distribution grant:
+        
+            Your application must have material additional functionality, beyond the included portions of the SDK.
+        
+            The distributable portions of the SDK shall only be accessed by your application.
+        
+            The following notice shall be included in modifications and derivative works of sample source code distributed: “This software contains source code provided by NVIDIA Corporation.”
+        
+            Unless a developer tool is identified in this Agreement as distributable, it is delivered for your internal use only.
+        
+            The terms under which you distribute your application must be consistent with the terms of this Agreement, including (without limitation) terms relating to the license grant and license restrictions and protection of NVIDIA’s intellectual property rights. Additionally, you agree that you will protect the privacy, security and legal rights of your application users.
+        
+            You agree to notify NVIDIA in writing of any known or suspected distribution or use of the SDK not in compliance with the requirements of this Agreement, and to enforce the terms of your agreements with respect to distributed SDK.
+        
+        1.3 Authorized Users
+        
+        You may allow employees and contractors of your entity or of your subsidiary(ies) to access and use the SDK from your secure network to perform work on your behalf.
+        
+        If you are an academic institution you may allow users enrolled or employed by the academic institution to access and use the SDK from your secure network.
+        
+        You are responsible for the compliance with the terms of this Agreement by your authorized users. If you become aware that your authorized users didn’t follow the terms of this Agreement, you agree to take reasonable steps to resolve the non-compliance and prevent new occurrences.
+        
+        1.4 Pre-Release SDK
+        
+        The SDK versions identified as alpha, beta, preview or otherwise as pre-release, may not be fully functional, may contain errors or design flaws, and may have reduced or different security, privacy, accessibility, availability, and reliability standards relative to commercial versions of NVIDIA software and materials. Use of a pre-release SDK may result in unexpected results, loss of data, project delays or other unpredictable damage or loss. You may use a pre-release SDK at your own risk, understanding that pre-release SDKs are not intended for use in production or business-critical systems. NVIDIA may choose not to make available a commercial version of any pre-release SDK. NVIDIA may also choose to abandon development and terminate the availability of a pre-release SDK at any time without liability. 1.5 Updates
+        
+        NVIDIA may, at its option, make available patches, workarounds or other updates to this SDK. Unless the updates are provided with their separate governing terms, they are deemed part of the SDK licensed to you as provided in this Agreement.
+        
+        You agree that the form and content of the SDK that NVIDIA provides may change without prior notice to you. While NVIDIA generally maintains compatibility between versions, NVIDIA may in some cases make changes that introduce incompatibilities in future versions of the SDK.
+        
+        1.6 Components Under Other Licenses.
+        
+        The SDK may include NVIDIA or third-party components with separate legal notices or terms as may be described in proprietary notices accompanying the SDK, such as components governed by open source software licenses. If and to the extent there is a conflict between the terms in this license and the license terms associated with a component, the license terms associated with the components control only to the extent necessary to resolve the conflict.
+        
+        1.7 Reservation of Rights
+        
+        NVIDIA reserves all rights, title and interest in and to the SDK not expressly granted to you under this Agreement.
+        
+            Limitations.
+        
+        The following license limitations apply to your use of the SDK:
+        
+        2.1 The SDK is licensed for you to develop applications only for use in systems with NVIDIA GPUs.
+        
+        2.2 You may not reverse engineer, decompile or disassemble, or remove copyright or other proprietary notices from any portion of the SDK or copies of the SDK.
+        
+        2.3 Except as expressly provided in this Agreement, you may not copy, sell, rent, sublicense, transfer, distribute, modify, or create derivative works of any portion of the SDK.
+        
+        2.4 Unless you have an agreement with NVIDIA for this purpose, you may not indicate that an application created with the SDK is sponsored or endorsed by NVIDIA.
+        
+        2.5 You may not bypass, disable, or circumvent any encryption, security, digital rights management or authentication mechanism in the SDK.
+        
+        2.6 You may not use the SDK in any manner that would cause it to become subject to an open source software license. As examples, licenses that require as a condition of use, modification, and/or distribution that the SDK be (i) disclosed or distributed in source code form; (ii) licensed for the purpose of making derivative works; or (iii) redistributable at no charge.
+        
+        2.7 You acknowledge that the SDK as delivered is not tested or certified by NVIDIA for use in connection with the design, construction, maintenance, and/or operation of any system where the use or failure of such system could result in a situation that threatens the safety of human life or results in catastrophic damages (each, a “Critical Application”). Examples of Critical Applications include use in avionics, navigation, autonomous vehicle applications, ai solutions for automotive products, military, medical, life support or other life critical applications. NVIDIA shall not be liable to you or any third party, in whole or in part, for any claims or damages arising from such uses. You are solely responsible for ensuring that any product or service developed with the SDK as a whole includes sufficient features to comply with all applicable legal and regulatory standards and requirements.
+        
+        2.8 You agree to defend, indemnify and hold harmless NVIDIA and its affiliates, and their respective employees, contractors, agents, officers and directors, from and against any and all claims, damages, obligations, losses, liabilities, costs or debt, fines, restitutions and expenses (including but not limited to attorney’s fees and costs incident to establishing the right of indemnification) arising out of or related to products or services that use the SDK in or for Critical Applications, and for use of the SDK outside of the scope of this Agreement, or not in compliance with its terms.
+        
+            Ownership.
+        
+        3.1 NVIDIA or its licensors hold all rights, title and interest in and to the SDK and its modifications, including their respective intellectual property rights. This SDK may include software and materials from NVIDIA’s licensors, and these licensors are intended third party beneficiaries that may enforce this Agreement with respect to their intellectual property rights.
+        
+        3.2 You may, but don’t have to, provide to NVIDIA suggestions, feature requests or other feedback regarding the SDK, including possible enhancements or modifications to the SDK. For any feedback that you voluntarily provide, you hereby grant NVIDIA and its affiliates a perpetual, non-exclusive, worldwide, irrevocable license to use, reproduce, modify, license, sublicense (through multiple tiers of sublicensees), and distribute (through multiple tiers of distributors) it without the payment of any royalties or fees to you. NVIDIA will use feedback at its choice.
+        
+            No Warranties.
+        
+        THE SDK IS PROVIDED BY NVIDIA “AS IS” AND “WITH ALL FAULTS.” TO THE MAXIMUM EXTENT PERMITTED BY LAW, NVIDIA AND ITS AFFILIATES EXPRESSLY DISCLAIM ALL WARRANTIES OF ANY KIND OR NATURE, WHETHER EXPRESS, IMPLIED OR STATUTORY, INCLUDING, BUT NOT LIMITED TO, ANY WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, TITLE, NON-INFRINGEMENT, OR THE ABSENCE OF ANY DEFECTS THEREIN, WHETHER LATENT OR PATENT. NO WARRANTY IS MADE ON THE BASIS OF TRADE USAGE, COURSE OF DEALING OR COURSE OF TRADE.
+        
+            Limitations of Liability.
+        
+        TO THE MAXIMUM EXTENT PERMITTED BY LAW, NVIDIA AND ITS AFFILIATES SHALL NOT BE LIABLE FOR ANY SPECIAL, INCIDENTAL, PUNITIVE OR CONSEQUENTIAL DAMAGES, OR ANY LOST PROFITS, LOSS OF USE, LOSS OF DATA OR LOSS OF GOODWILL, OR THE COSTS OF PROCURING SUBSTITUTE PRODUCTS, ARISING OUT OF OR IN CONNECTION WITH THIS AGREEMENT OR THE USE OR PERFORMANCE OF THE SDK, WHETHER SUCH LIABILITY ARISES FROM ANY CLAIM BASED UPON BREACH OF CONTRACT, BREACH OF WARRANTY, TORT (INCLUDING NEGLIGENCE), PRODUCT LIABILITY OR ANY OTHER CAUSE OF ACTION OR THEORY OF LIABILITY. IN NO EVENT WILL NVIDIA’S AND ITS AFFILIATES TOTAL CUMULATIVE LIABILITY UNDER OR ARISING OUT OF THIS AGREEMENT EXCEED US$10.00. THE NATURE OF THE LIABILITY OR THE NUMBER OF CLAIMS OR SUITS SHALL NOT ENLARGE OR EXTEND THIS LIMIT.
+        
+        These exclusions and limitations of liability shall apply regardless if NVIDIA or its affiliates have been advised of the possibility of such damages, and regardless of whether a remedy fails its essential purpose. These exclusions and limitations of liability form an essential basis of the bargain between the parties, and, absent any of these exclusions or limitations of liability, the provisions of this Agreement, including, without limitation, the economic terms, would be substantially different.
+        
+            Termination.
+        
+        6.1 This Agreement will continue to apply until terminated by either you or NVIDIA as described below.
+        
+        6.2 If you want to terminate this Agreement, you may do so by stopping to use the SDK.
+        
+        6.3 NVIDIA may, at any time, terminate this Agreement if: (i) you fail to comply with any term of this Agreement and the non-compliance is not fixed within thirty (30) days following notice from NVIDIA (or immediately if you violate NVIDIA’s intellectual property rights); (ii) you commence or participate in any legal proceeding against NVIDIA with respect to the SDK; or (iii) NVIDIA decides to no longer provide the SDK in a country or, in NVIDIA’s sole discretion, the continued use of it is no longer commercially viable.
+        
+        6.4 Upon any termination of this Agreement, you agree to promptly discontinue use of the SDK and destroy all copies in your possession or control. Your prior distributions in accordance with this Agreement are not affected by the termination of this Agreement. Upon written request, you will certify in writing that you have complied with your commitments under this section. Upon any termination of this Agreement all provisions survive except for the licenses granted to you.
+        
+            General.
+        
+        If you wish to assign this Agreement or your rights and obligations, including by merger, consolidation, dissolution or operation of law, contact NVIDIA to ask for permission. Any attempted assignment not approved by NVIDIA in writing shall be void and of no effect. NVIDIA may assign, delegate or transfer this Agreement and its rights and obligations, and if to a non-affiliate you will be notified.
+        
+        You agree to cooperate with NVIDIA and provide reasonably requested information to verify your compliance with this Agreement.
+        
+        This Agreement will be governed in all respects by the laws of the United States and of the State of Delaware as those laws are applied to contracts entered into and performed entirely within Delaware by Delaware residents, without regard to the conflicts of laws principles. The United Nations Convention on Contracts for the International Sale of Goods is specifically disclaimed. You agree to all terms of this Agreement in the English language.
+        
+        The state or federal courts residing in Santa Clara County, California shall have exclusive jurisdiction over any dispute or claim arising out of this Agreement. Notwithstanding this, you agree that NVIDIA shall still be allowed to apply for injunctive remedies or an equivalent type of urgent legal relief in any jurisdiction.
+        
+        If any court of competent jurisdiction determines that any provision of this Agreement is illegal, invalid or unenforceable, such provision will be construed as limited to the extent necessary to be consistent with and fully enforceable under the law and the remaining provisions will remain in full force and effect. Unless otherwise specified, remedies are cumulative.
+        
+        Each party acknowledges and agrees that the other is an independent contractor in the performance of this Agreement.
+        
+        The SDK has been developed entirely at private expense and is “commercial items” consisting of “commercial computer software” and “commercial computer software documentation” provided with RESTRICTED RIGHTS. Use, duplication or disclosure by the U.S. Government or a U.S. Government subcontractor is subject to the restrictions in this Agreement pursuant to DFARS 227.7202-3(a) or as set forth in subparagraphs (b)(1) and (2) of the Commercial Computer Software - Restricted Rights clause at FAR 52.227-19, as applicable. Contractor/manufacturer is NVIDIA, 2788 San Tomas Expressway, Santa Clara, CA 95051.
+        
+        The SDK is subject to United States export laws and regulations. You agree that you will not ship, transfer or export the SDK into any country, or use the SDK in any manner, prohibited by the United States Bureau of Industry and Security or economic sanctions regulations administered by the U.S. Department of Treasury’s Office of Foreign Assets Control (OFAC), or any applicable export laws, restrictions or regulations. These laws include restrictions on destinations, end users and end use. By accepting this Agreement, you confirm that you are not a resident or citizen of any country currently embargoed by the U.S. and that you are not otherwise prohibited from receiving the SDK.
+        
+        Any notice delivered by NVIDIA to you under this Agreement will be delivered via mail, email or fax. You agree that any notices that NVIDIA sends you electronically will satisfy any legal communication requirements. Please direct your legal notices or other correspondence to NVIDIA Corporation, 2788 San Tomas Expressway, Santa Clara, California 95051, United States of America, Attention: Legal Department.
+        
+        This Agreement constitutes the entire agreement of the parties with respect to the subject matter of this Agreement and supersedes all prior negotiations or documentation exchanged between the parties relating to this subject matter. Any additional and/or conflicting terms on documents issued by you are null, void, and invalid. Any amendment or waiver under this Agreement shall be in writing and signed by representatives of both parties.
+        
+        If the distribution terms in this Agreement are not suitable for your organization, or for any questions regarding this Agreement, please contact NVIDIA at nvidia-compute-license-questions@nvidia.com.
+        
+        (v. February 10, 2022)
+        
+        # Third Party License Agreements
+        
+        ## Nanobind
+        
+        Copyright (c) 2022 Wenzel Jakob <wenzel.jakob@epfl.ch>, All rights reserved.
+        
+        Redistribution and use in source and binary forms, with or without
+        modification, are permitted provided that the following conditions are met:
+        
+        1. Redistributions of source code must retain the above copyright notice, this
+           list of conditions and the following disclaimer.
+        
+        2. Redistributions in binary form must reproduce the above copyright notice,
+           this list of conditions and the following disclaimer in the documentation
+           and/or other materials provided with the distribution.
+        
+        3. Neither the name of the copyright holder nor the names of its contributors
+           may be used to endorse or promote products derived from this software
+           without specific prior written permission.
+        
+        THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
+        ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+        WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+        DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+        FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+        DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+        SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+        CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+        OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+        OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+Classifier: Intended Audience :: Developers
+Classifier: Programming Language :: Python
+Project-URL: Homepage, https://github.com/nvidia/cuEquivariance
+Project-URL: Documentation, https://github.com/nvidia/cuEquivariance
+Requires-Python: >=3.10
+Provides-Extra: test
+Requires-Dist: numpy; extra == "test"
+Requires-Dist: pytest; extra == "test"
+Description-Content-Type: text/markdown
+
+# cuequivariance-ops
+
+
+## Introduction
+
+`cuequivariance_ops` provides CUDA kernels for the
+[cuEquivariance](https://github.com/nvidia/cuEquivariance).
+This python module does not contain any python bindings.
+When imported, it loads the shared library that contains the CUDA kernels.
+
+
+## Documentation
+
+Refer to
+[cuEquivariance](https://github.com/nvidia/cuEquivariance).
+
+
+## Support and Feedback
+
+Please contact the cuEquivariance developers for any issues you might encounter.

cuequivariance_ops_cu12-0.4.0.dist-info/RECORD

@@ -0,0 +1,13 @@
+cuequivariance_ops_cu12-0.4.0.dist-info/RECORD,,
+cuequivariance_ops_cu12-0.4.0.dist-info/WHEEL,sha256=irdNKpSXnQRqSJjb2WbcCyjS-JSpVoNucdwzaZi1ofk,151
+cuequivariance_ops_cu12-0.4.0.dist-info/METADATA,sha256=qKROsUuxgmCYP0N6r9qb0RtpKKXS9MAKvEqzvmmnDXE,20665
+cuequivariance_ops_cu12-0.4.0.dist-info/licenses/LICENSE,sha256=rvp0QV9FuOdxz_CGWTd9DgId4xh2BByyXfBBnb0ejZM,18279
+cuequivariance_ops/__init__.py,sha256=ba7jv_WICRROtLbDU2O1u0MHxp6VkVu0-UGKuQxf9iw,1255
+cuequivariance_ops/VERSION,sha256=QLjrQACpE6d5EJBTXykdPTaYdBYqie88nj1OiHobnnk,6
+cuequivariance_ops/_version.py,sha256=o9Flao_mTq2Y7TrrjnSCqEAgebmA0sGozsl15qVI13Y,730
+cuequivariance_ops/common/nvtx.hpp,sha256=Wi6z9b-yFUNq6ShJjjcsdxQRqCygd4xGegGJrqUI9Wk,708
+cuequivariance_ops/common/common.hpp,sha256=2zDyE5lGugQL43vmM4_ylmp-Tz8OBFnPRsdFra_1BdM,2787
+cuequivariance_ops/lib/libcue_ops.so,sha256=TqrkFt7uGcnZj2ddnqXDgvOuwPlbJ_3ju5IJKbHyDOY,84613176
+cuequivariance_ops/equivariance/tensor_product_uniform_1d_jit.hh,sha256=oWhSS0ZmMHlye8eTucweoGBtzN1H0nN1GX_Rz-MsPqI,2002
+cuequivariance_ops/equivariance/fused_tensor_product.cuh,sha256=bOXR5UWU9gNYRfdh6k28NEkV3CUU2ijmh6y7c0ND0J4,8283
+cuequivariance_ops/equivariance/segmented_transpose.cuh,sha256=gfSZhRBwSqwVAgFCCiGtI-NJ8yDy9tV_iCg1G2KpctY,1766

cuequivariance_ops_cu12-0.4.0.dist-info/WHEEL

@@ -0,0 +1,6 @@
+Wheel-Version: 1.0
+Generator: scikit-build-core 0.10.6
+Root-Is-Purelib: false
+Tag: py3-none-manylinux_2_24_x86_64
+Tag: py3-none-manylinux_2_28_x86_64
+

cuequivariance_ops_cu12-0.4.0.dist-info/licenses/LICENSE

@@ -0,0 +1,168 @@
+# Software License Agreement
+
+LICENSE AGREEMENT FOR NVIDIA MATH LIBRARIES SOFTWARE DEVELOPMENT KITS
+
+This license agreement(“Agreement”) is a legal agreement between you and NVIDIA Corporation (“NVIDIA”) and governs your use of the NVIDIA math libraries software development kit as available at NVIDIA’s discretion (each, a “SDK”).
+
+Each SDK has its own set of software and materials, but here is a description of the types of items that may be included in a SDK: source code, header files, APIs, data sets and assets (examples include images, textures, models, scenes, videos, native API input/output files), binary software, sample code, libraries, utility programs, programming code and documentation.
+
+This Agreement can be accepted only by an adult of legal age of majority in the country in which the SDK is used.
+
+If you are entering into this Agreement on behalf of a company or other legal entity, you represent that you have the legal authority to bind the entity to this Agreement, in which case “you” will mean the entity you represent.
+
+If you don’t have the required age or authority to accept this Agreement, or if you don’t accept all the terms and conditions of this Agreement, do not download, install or use the SDK.
+
+You agree to use the SDK only for purposes that are permitted by (a) this Agreement, and (b) any applicable law, regulation or generally accepted practices or guidelines in the relevant jurisdictions.
+
+    License.
+
+1.1 Grant
+
+Subject to the terms of this Agreement, NVIDIA hereby grants you a non-exclusive, non-transferable license, without the right to sublicense (except as expressly provided in this Agreement) to:
+
+    Install and use the SDK, and
+
+    Distribute the binary files, files identified as samples, and headers as incorporated into a software application that meets the distribution requirements indicated in this Agreement.
+
+1.2 Distribution Requirements
+
+These are the distribution requirements for you to exercise the distribution grant:
+
+    Your application must have material additional functionality, beyond the included portions of the SDK.
+
+    The distributable portions of the SDK shall only be accessed by your application.
+
+    The following notice shall be included in modifications and derivative works of sample source code distributed: “This software contains source code provided by NVIDIA Corporation.”
+
+    Unless a developer tool is identified in this Agreement as distributable, it is delivered for your internal use only.
+
+    The terms under which you distribute your application must be consistent with the terms of this Agreement, including (without limitation) terms relating to the license grant and license restrictions and protection of NVIDIA’s intellectual property rights. Additionally, you agree that you will protect the privacy, security and legal rights of your application users.
+
+    You agree to notify NVIDIA in writing of any known or suspected distribution or use of the SDK not in compliance with the requirements of this Agreement, and to enforce the terms of your agreements with respect to distributed SDK.
+
+1.3 Authorized Users
+
+You may allow employees and contractors of your entity or of your subsidiary(ies) to access and use the SDK from your secure network to perform work on your behalf.
+
+If you are an academic institution you may allow users enrolled or employed by the academic institution to access and use the SDK from your secure network.
+
+You are responsible for the compliance with the terms of this Agreement by your authorized users. If you become aware that your authorized users didn’t follow the terms of this Agreement, you agree to take reasonable steps to resolve the non-compliance and prevent new occurrences.
+
+1.4 Pre-Release SDK
+
+The SDK versions identified as alpha, beta, preview or otherwise as pre-release, may not be fully functional, may contain errors or design flaws, and may have reduced or different security, privacy, accessibility, availability, and reliability standards relative to commercial versions of NVIDIA software and materials. Use of a pre-release SDK may result in unexpected results, loss of data, project delays or other unpredictable damage or loss. You may use a pre-release SDK at your own risk, understanding that pre-release SDKs are not intended for use in production or business-critical systems. NVIDIA may choose not to make available a commercial version of any pre-release SDK. NVIDIA may also choose to abandon development and terminate the availability of a pre-release SDK at any time without liability. 1.5 Updates
+
+NVIDIA may, at its option, make available patches, workarounds or other updates to this SDK. Unless the updates are provided with their separate governing terms, they are deemed part of the SDK licensed to you as provided in this Agreement.
+
+You agree that the form and content of the SDK that NVIDIA provides may change without prior notice to you. While NVIDIA generally maintains compatibility between versions, NVIDIA may in some cases make changes that introduce incompatibilities in future versions of the SDK.
+
+1.6 Components Under Other Licenses.
+
+The SDK may include NVIDIA or third-party components with separate legal notices or terms as may be described in proprietary notices accompanying the SDK, such as components governed by open source software licenses. If and to the extent there is a conflict between the terms in this license and the license terms associated with a component, the license terms associated with the components control only to the extent necessary to resolve the conflict.
+
+1.7 Reservation of Rights
+
+NVIDIA reserves all rights, title and interest in and to the SDK not expressly granted to you under this Agreement.
+
+    Limitations.
+
+The following license limitations apply to your use of the SDK:
+
+2.1 The SDK is licensed for you to develop applications only for use in systems with NVIDIA GPUs.
+
+2.2 You may not reverse engineer, decompile or disassemble, or remove copyright or other proprietary notices from any portion of the SDK or copies of the SDK.
+
+2.3 Except as expressly provided in this Agreement, you may not copy, sell, rent, sublicense, transfer, distribute, modify, or create derivative works of any portion of the SDK.
+
+2.4 Unless you have an agreement with NVIDIA for this purpose, you may not indicate that an application created with the SDK is sponsored or endorsed by NVIDIA.
+
+2.5 You may not bypass, disable, or circumvent any encryption, security, digital rights management or authentication mechanism in the SDK.
+
+2.6 You may not use the SDK in any manner that would cause it to become subject to an open source software license. As examples, licenses that require as a condition of use, modification, and/or distribution that the SDK be (i) disclosed or distributed in source code form; (ii) licensed for the purpose of making derivative works; or (iii) redistributable at no charge.
+
+2.7 You acknowledge that the SDK as delivered is not tested or certified by NVIDIA for use in connection with the design, construction, maintenance, and/or operation of any system where the use or failure of such system could result in a situation that threatens the safety of human life or results in catastrophic damages (each, a “Critical Application”). Examples of Critical Applications include use in avionics, navigation, autonomous vehicle applications, ai solutions for automotive products, military, medical, life support or other life critical applications. NVIDIA shall not be liable to you or any third party, in whole or in part, for any claims or damages arising from such uses. You are solely responsible for ensuring that any product or service developed with the SDK as a whole includes sufficient features to comply with all applicable legal and regulatory standards and requirements.
+
+2.8 You agree to defend, indemnify and hold harmless NVIDIA and its affiliates, and their respective employees, contractors, agents, officers and directors, from and against any and all claims, damages, obligations, losses, liabilities, costs or debt, fines, restitutions and expenses (including but not limited to attorney’s fees and costs incident to establishing the right of indemnification) arising out of or related to products or services that use the SDK in or for Critical Applications, and for use of the SDK outside of the scope of this Agreement, or not in compliance with its terms.
+
+    Ownership.
+
+3.1 NVIDIA or its licensors hold all rights, title and interest in and to the SDK and its modifications, including their respective intellectual property rights. This SDK may include software and materials from NVIDIA’s licensors, and these licensors are intended third party beneficiaries that may enforce this Agreement with respect to their intellectual property rights.
+
+3.2 You may, but don’t have to, provide to NVIDIA suggestions, feature requests or other feedback regarding the SDK, including possible enhancements or modifications to the SDK. For any feedback that you voluntarily provide, you hereby grant NVIDIA and its affiliates a perpetual, non-exclusive, worldwide, irrevocable license to use, reproduce, modify, license, sublicense (through multiple tiers of sublicensees), and distribute (through multiple tiers of distributors) it without the payment of any royalties or fees to you. NVIDIA will use feedback at its choice.
+
+    No Warranties.
+
+THE SDK IS PROVIDED BY NVIDIA “AS IS” AND “WITH ALL FAULTS.” TO THE MAXIMUM EXTENT PERMITTED BY LAW, NVIDIA AND ITS AFFILIATES EXPRESSLY DISCLAIM ALL WARRANTIES OF ANY KIND OR NATURE, WHETHER EXPRESS, IMPLIED OR STATUTORY, INCLUDING, BUT NOT LIMITED TO, ANY WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, TITLE, NON-INFRINGEMENT, OR THE ABSENCE OF ANY DEFECTS THEREIN, WHETHER LATENT OR PATENT. NO WARRANTY IS MADE ON THE BASIS OF TRADE USAGE, COURSE OF DEALING OR COURSE OF TRADE.
+
+    Limitations of Liability.
+
+TO THE MAXIMUM EXTENT PERMITTED BY LAW, NVIDIA AND ITS AFFILIATES SHALL NOT BE LIABLE FOR ANY SPECIAL, INCIDENTAL, PUNITIVE OR CONSEQUENTIAL DAMAGES, OR ANY LOST PROFITS, LOSS OF USE, LOSS OF DATA OR LOSS OF GOODWILL, OR THE COSTS OF PROCURING SUBSTITUTE PRODUCTS, ARISING OUT OF OR IN CONNECTION WITH THIS AGREEMENT OR THE USE OR PERFORMANCE OF THE SDK, WHETHER SUCH LIABILITY ARISES FROM ANY CLAIM BASED UPON BREACH OF CONTRACT, BREACH OF WARRANTY, TORT (INCLUDING NEGLIGENCE), PRODUCT LIABILITY OR ANY OTHER CAUSE OF ACTION OR THEORY OF LIABILITY. IN NO EVENT WILL NVIDIA’S AND ITS AFFILIATES TOTAL CUMULATIVE LIABILITY UNDER OR ARISING OUT OF THIS AGREEMENT EXCEED US$10.00. THE NATURE OF THE LIABILITY OR THE NUMBER OF CLAIMS OR SUITS SHALL NOT ENLARGE OR EXTEND THIS LIMIT.
+
+These exclusions and limitations of liability shall apply regardless if NVIDIA or its affiliates have been advised of the possibility of such damages, and regardless of whether a remedy fails its essential purpose. These exclusions and limitations of liability form an essential basis of the bargain between the parties, and, absent any of these exclusions or limitations of liability, the provisions of this Agreement, including, without limitation, the economic terms, would be substantially different.
+
+    Termination.
+
+6.1 This Agreement will continue to apply until terminated by either you or NVIDIA as described below.
+
+6.2 If you want to terminate this Agreement, you may do so by stopping to use the SDK.
+
+6.3 NVIDIA may, at any time, terminate this Agreement if: (i) you fail to comply with any term of this Agreement and the non-compliance is not fixed within thirty (30) days following notice from NVIDIA (or immediately if you violate NVIDIA’s intellectual property rights); (ii) you commence or participate in any legal proceeding against NVIDIA with respect to the SDK; or (iii) NVIDIA decides to no longer provide the SDK in a country or, in NVIDIA’s sole discretion, the continued use of it is no longer commercially viable.
+
+6.4 Upon any termination of this Agreement, you agree to promptly discontinue use of the SDK and destroy all copies in your possession or control. Your prior distributions in accordance with this Agreement are not affected by the termination of this Agreement. Upon written request, you will certify in writing that you have complied with your commitments under this section. Upon any termination of this Agreement all provisions survive except for the licenses granted to you.
+
+    General.
+
+If you wish to assign this Agreement or your rights and obligations, including by merger, consolidation, dissolution or operation of law, contact NVIDIA to ask for permission. Any attempted assignment not approved by NVIDIA in writing shall be void and of no effect. NVIDIA may assign, delegate or transfer this Agreement and its rights and obligations, and if to a non-affiliate you will be notified.
+
+You agree to cooperate with NVIDIA and provide reasonably requested information to verify your compliance with this Agreement.
+
+This Agreement will be governed in all respects by the laws of the United States and of the State of Delaware as those laws are applied to contracts entered into and performed entirely within Delaware by Delaware residents, without regard to the conflicts of laws principles. The United Nations Convention on Contracts for the International Sale of Goods is specifically disclaimed. You agree to all terms of this Agreement in the English language.
+
+The state or federal courts residing in Santa Clara County, California shall have exclusive jurisdiction over any dispute or claim arising out of this Agreement. Notwithstanding this, you agree that NVIDIA shall still be allowed to apply for injunctive remedies or an equivalent type of urgent legal relief in any jurisdiction.
+
+If any court of competent jurisdiction determines that any provision of this Agreement is illegal, invalid or unenforceable, such provision will be construed as limited to the extent necessary to be consistent with and fully enforceable under the law and the remaining provisions will remain in full force and effect. Unless otherwise specified, remedies are cumulative.
+
+Each party acknowledges and agrees that the other is an independent contractor in the performance of this Agreement.
+
+The SDK has been developed entirely at private expense and is “commercial items” consisting of “commercial computer software” and “commercial computer software documentation” provided with RESTRICTED RIGHTS. Use, duplication or disclosure by the U.S. Government or a U.S. Government subcontractor is subject to the restrictions in this Agreement pursuant to DFARS 227.7202-3(a) or as set forth in subparagraphs (b)(1) and (2) of the Commercial Computer Software - Restricted Rights clause at FAR 52.227-19, as applicable. Contractor/manufacturer is NVIDIA, 2788 San Tomas Expressway, Santa Clara, CA 95051.
+
+The SDK is subject to United States export laws and regulations. You agree that you will not ship, transfer or export the SDK into any country, or use the SDK in any manner, prohibited by the United States Bureau of Industry and Security or economic sanctions regulations administered by the U.S. Department of Treasury’s Office of Foreign Assets Control (OFAC), or any applicable export laws, restrictions or regulations. These laws include restrictions on destinations, end users and end use. By accepting this Agreement, you confirm that you are not a resident or citizen of any country currently embargoed by the U.S. and that you are not otherwise prohibited from receiving the SDK.
+
+Any notice delivered by NVIDIA to you under this Agreement will be delivered via mail, email or fax. You agree that any notices that NVIDIA sends you electronically will satisfy any legal communication requirements. Please direct your legal notices or other correspondence to NVIDIA Corporation, 2788 San Tomas Expressway, Santa Clara, California 95051, United States of America, Attention: Legal Department.
+
+This Agreement constitutes the entire agreement of the parties with respect to the subject matter of this Agreement and supersedes all prior negotiations or documentation exchanged between the parties relating to this subject matter. Any additional and/or conflicting terms on documents issued by you are null, void, and invalid. Any amendment or waiver under this Agreement shall be in writing and signed by representatives of both parties.
+
+If the distribution terms in this Agreement are not suitable for your organization, or for any questions regarding this Agreement, please contact NVIDIA at nvidia-compute-license-questions@nvidia.com.
+
+(v. February 10, 2022)
+
+# Third Party License Agreements
+
+## Nanobind
+
+Copyright (c) 2022 Wenzel Jakob <wenzel.jakob@epfl.ch>, All rights reserved.
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are met:
+
+1. Redistributions of source code must retain the above copyright notice, this
+   list of conditions and the following disclaimer.
+
+2. Redistributions in binary form must reproduce the above copyright notice,
+   this list of conditions and the following disclaimer in the documentation
+   and/or other materials provided with the distribution.
+
+3. Neither the name of the copyright holder nor the names of its contributors
+   may be used to endorse or promote products derived from this software
+   without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
+ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.