mplang-nightly 0.1.dev268__py3-none-any.whl → 0.1.dev270__py3-none-any.whl

mplang/{v2/runtime → runtime}/interpreter.py

@@ -33,16 +33,16 @@ import time
 from collections.abc import Callable
 from typing import TYPE_CHECKING, Any, cast
 
-from mplang.v2.edsl.context import AbstractInterpreter
-from mplang.v2.edsl.graph import Graph
-from mplang.v2.edsl.object import Object
-from mplang.v2.edsl.registry import get_impl
-from mplang.v2.edsl.typing import BaseType
-from mplang.v2.runtime.dialect_state import DialectState
-from mplang.v2.runtime.object_store import ObjectStore
+from mplang.edsl.context import AbstractInterpreter
+from mplang.edsl.graph import Graph
+from mplang.edsl.object import Object
+from mplang.edsl.registry import get_impl
+from mplang.edsl.typing import BaseType
+from mplang.runtime.dialect_state import DialectState
+from mplang.runtime.object_store import ObjectStore
 
 if TYPE_CHECKING:
-    from mplang.v2.edsl.primitive import Primitive
+    from mplang.edsl.primitive import Primitive
 
 
 class ExecutionTracer:
@@ -442,8 +442,8 @@ class Interpreter(AbstractInterpreter):
         Returns:
             Execution result (InterpObject or list of InterpObject or mixed with immediates)
         """
-        from mplang.v1.utils.func_utils import var_demorph, var_morph
-        from mplang.v2.edsl.tracer import Tracer
+        from mplang.edsl.tracer import Tracer
+        from mplang.utils import var_demorph, var_morph
 
         # Create tracer and build graph
         # Note: primitive.bind() internally calls Tracer.lift() with is_param=False,
@@ -525,7 +525,7 @@ class Interpreter(AbstractInterpreter):
             >>> interp = Interpreter()
             >>> z_interp = interp.lift(z_trace)  # evaluate graph → InterpObject
         """
-        from mplang.v2.edsl.tracer import TraceObject
+        from mplang.edsl.tracer import TraceObject
 
         if isinstance(obj, InterpObject):
             # InterpObject must belong to this interpreter

mplang/{v2/runtime → runtime}/value.py

@@ -22,8 +22,8 @@ This module defines:
    implements the "wrap/unwrap" pattern with automatic type conversion.
 
 Usage:
-    from mplang.v2.runtime.value import Value
-    from mplang.v2.edsl import serde
+    from mplang.runtime.value import Value
+    from mplang.edsl import serde
 
     @serde.register_class
     class MyValue(Value):

mplang/{v1/runtime → utils}/__init__.py

@@ -12,21 +12,24 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-"""
-Runtime components for mplang.
-
-This module contains runtime implementations including:
-- Simulator for local simulation
-- Driver for distributed execution
-"""
-
-from mplang.v1.runtime.driver import Driver, DriverVar
-from mplang.v1.runtime.simulation import Simulator
+from .func_utils import (
+    MorphStruct,
+    is_treedef_list,
+    list_reconstruct,
+    list_split,
+    normalize_fn,
+    validate_morph_struct,
+    var_demorph,
+    var_morph,
+)
 
 __all__ = [
-    "Driver",
-    "DriverVar",
-    "Simulator",
-    # "serve",
-    # "start_cluster",
+    "MorphStruct",
+    "is_treedef_list",
+    "list_reconstruct",
+    "list_split",
+    "normalize_fn",
+    "validate_morph_struct",
+    "var_demorph",
+    "var_morph",
 ]

mplang/{v1/utils → utils}/func_utils.py

@@ -119,7 +119,7 @@ def normalize_fn(
     return normalized, params
 
 
-def is_treedef_list(treedef: PyTreeDef) -> bool:
+def is_treedef_list(treedef: PyTreeDef) -> bool:  # type: ignore
     """
     Checks if a PyTreeDef object represents a simple (non-nested) list.
 

{mplang_nightly-0.1.dev268.dist-info → mplang_nightly-0.1.dev270.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mplang-nightly
-Version: 0.1.dev268
+Version: 0.1.dev270
 Summary: Multi-Party Programming Language
 Author-email: SecretFlow Team <secretflow-contact@service.alipay.com>
 License:                                  Apache License
@@ -214,7 +214,7 @@ Requires-Dist: flax>=0.12.0
 Requires-Dist: httpx<1.0.0,>=0.27.0
 Requires-Dist: jax[cpu]==0.8.0
 Requires-Dist: lightphe<0.1.0,>=0.0.15
-Requires-Dist: numpy>=2.0.0
+Requires-Dist: numpy!=2.4.0,>=2.0.0
 Requires-Dist: pandas>=2.0.0
 Requires-Dist: protobuf<6.0,>=5.0
 Requires-Dist: pyarrow>=14.0.0

mplang_nightly-0.1.dev270.dist-info/RECORD

@@ -0,0 +1,102 @@
+mplang/__init__.py,sha256=q-H8IXjVuI79eHhSFVh4FT5KbYow4lGyji2MyLsjufw,13429
+mplang/cli.py,sha256=PqxgWL0rtG-0zuip4JzoVKTIQ5gAwIwHzG3xlZJaBwI,19638
+mplang/cli_guide.md,sha256=hKC6AKgJn-lM_wZ0CzZIP2QUBxGPnT0Op_1YyeUhCfI,3581
+mplang/py.typed,sha256=RyhZV7Yxo8BvEoBiFz5fH3IWVqQKTcBBE-bzjx_N5GQ,583
+mplang/backends/__init__.py,sha256=EnKHmOX4YgvNVA42k_KfiS-Vn860yN7Rj9XkUU0g9ys,2042
+mplang/backends/bfv_impl.py,sha256=qobGXe7ldWvYVcqdkNsTcVdNV-hUVE9-Ks5pd3X0Mvc,25691
+mplang/backends/channel.py,sha256=t8P7RphNnhvo3Zj08hX85d7PxgWQXiMGNYfZPK_4YoM,6622
+mplang/backends/crypto_impl.py,sha256=m5tbZpedn4McG5t6HIaEjhTQUFOIaEDMdyEG0499S0Y,23187
+mplang/backends/field_impl.py,sha256=lIghppy0i8QL7aTNCr8tNWjC_bvOXBIKLnavaO0fIOA,14773
+mplang/backends/func_impl.py,sha256=vhkSJnvgSzvQWXqOE40O8npqOmtNqLxefzv9ETuQODQ,3573
+mplang/backends/phe_impl.py,sha256=CefRpFrRJvbZm3X58OYOFAyNZXKDp081Q6CT8fhdreE,4118
+mplang/backends/simp_design.md,sha256=u4YeDLKk5avsueXhzPtt4OUBToBmurOVK_BWTKTJ7w0,5246
+mplang/backends/spu_impl.py,sha256=6hGdEcZ34ZkwsD6ZnkrhNkmut_UnWWxNiOgHQwMKMCs,9451
+mplang/backends/spu_state.py,sha256=tgwOiWN6zfP-fBx2rN2cNnC-Y9GrSArqU-ZBT3NdXls,6557
+mplang/backends/store_impl.py,sha256=BdrCfNhFbcjswIvfxD2XKGYeBfBrmEQA3pj2-EDzFcI,2213
+mplang/backends/table_impl.py,sha256=-8KCVSboJkL81HPLRzTpUlSvC37d5qNLIQx1xluN3Pg,27542
+mplang/backends/tee_impl.py,sha256=5yed94Lr55hgSDT6j3FjdTboqzhl4HpJ9lGk1YGZBXQ,7026
+mplang/backends/tensor_impl.py,sha256=4-cypN0KuUD8dzpzAdWEWWuRlzLZwwyIap9QscWGLyk,18888
+mplang/backends/simp_driver/__init__.py,sha256=pQkHGR8HagnPB4OqagPZY-Ul19y6HjIJ5dY6QXZrBH8,1243
+mplang/backends/simp_driver/http.py,sha256=vrjfyEs1_TiMh7I-3cQhIwG6edE0R_eTGegvfFl5xx4,5526
+mplang/backends/simp_driver/mem.py,sha256=EAsDO4Zl2pHf8TY4K_whluhbAwaT1NhktBM9u4g5yYE,9101
+mplang/backends/simp_driver/ops.py,sha256=WYObWDRCsiXH0UBWZX5vD5W98ZPkd88U_qBV8SE5rA8,4503
+mplang/backends/simp_driver/state.py,sha256=dNmYMFN2D2BBdgs6C0YLaHrfaBRMgs05UNxMWw6tZIs,1713
+mplang/backends/simp_driver/values.py,sha256=Lz1utNSIzH-dCzZAEjU6JRcxPsfKGfUJrYl6gIuMOGw,1509
+mplang/backends/simp_worker/__init__.py,sha256=gdrSY1-MDkupCoJ8xwwH7em7fgVWv3J4gBJ45uHdzgg,961
+mplang/backends/simp_worker/http.py,sha256=qB62LsXTtokjpRqxd2f0hFpYO445pPaTrgI62BJjBqY,12626
+mplang/backends/simp_worker/mem.py,sha256=tMGiRppeca0TnY8WdqYQMQvsx5UVswCqdeOhiDlLQBs,3574
+mplang/backends/simp_worker/ops.py,sha256=X_mAhrSZkiN9GmgTJgJs1c50CFTNB_lo9zoPI4rIYU4,5510
+mplang/backends/simp_worker/state.py,sha256=nIu0ybvdYqRqp0TkoSneUF2u31evDHucCRduVBaDals,1445
+mplang/dialects/__init__.py,sha256=CYMmkeQVU0Znr9n3_5clZKb16u7acJ5jl5Zjbx4Tn1U,1478
+mplang/dialects/bfv.py,sha256=m5YfobFCBqn0lg2zBM9RNs2AC7i4PUQH2qXjHLHwSy4,22332
+mplang/dialects/crypto.py,sha256=oEVYyQZxUoVnLW0UN7TULU_06AcSJfBJ9PFupMnbOh4,22182
+mplang/dialects/dtypes.py,sha256=IzWBLpOumGJEy223oH_ECFKTXb1chhPAIebtxJeqLEA,12098
+mplang/dialects/field.py,sha256=s6fK3h0OCpnhQB6YaawEqjpI3TgHT6jMnJkTw3ltjAA,6371
+mplang/dialects/func.py,sha256=qt2H3ClIqdOU5T7cWxzCdoPoWqeXxmbqIsKT2b7w0Mg,4369
+mplang/dialects/phe.py,sha256=-5WvBnlhaUpClLvayKVQIKSof9ROXhZbv0tnz8CTSaY,22878
+mplang/dialects/simp.py,sha256=MA6zhlMMeiBD93yafZgBcQGadShqEATBB7R2VdyfMjQ,32851
+mplang/dialects/spu.py,sha256=zKIbFDcQsTtfVnPWLMR0rnp656Y8eWywDQIVqFSmVJ8,11353
+mplang/dialects/store.py,sha256=FkvTOdcNC_oCgJPMZfdpGLlKZAlelEqRIZ2WgMgYH-8,2108
+mplang/dialects/table.py,sha256=i9ruyh91_tSWu9rsLomrBUfqRdbHiZMMMJzNKfMrAUc,13534
+mplang/dialects/tee.py,sha256=BMFSbeK-Ck2jQP4qY9bZeNYTxEa7uEtUWLZLC4BPQxk,10111
+mplang/dialects/tensor.py,sha256=7aAYKaMaFjJ8N25yPFnmVhUuUdKJYy-M-a4NsZGE7kY,39893
+mplang/edsl/README.md,sha256=viflvdRojOa6Xk_UMRPqpuPGXcPGmdlv2-XR6LO7B58,7592
+mplang/edsl/__init__.py,sha256=ARYS7FkkSXwjWCsPLtWc9kL5OaR4wd5zCQhFTkzZUp0,2598
+mplang/edsl/context.py,sha256=Ln8n3bDe8_ISe42TAGzUuz8fw57-tu1APuihMfAtW1Y,10075
+mplang/edsl/graph.py,sha256=nCeCN7-bxfzyv40fmxcEXOaVUx14cOCaHfFb7A9OBnE,14968
+mplang/edsl/jit.py,sha256=Gm8qnaXW4_jhF3GVoenJgJmHw02D5KNaaXqxe_lfybk,2063
+mplang/edsl/object.py,sha256=dBl58q-ondjpjPNBh8zZvIEj6pJw2yEoz6TCaM_oleA,1906
+mplang/edsl/primitive.py,sha256=1I9A1Y7lsKgd-zBj2nte-jtCdcC6V0cHNhifvVsULkU,10560
+mplang/edsl/printer.py,sha256=drmfRkdCNqbkRfSDmejxtO-rEAaM13QyHB3AbAmKVFk,4393
+mplang/edsl/registry.py,sha256=hudXZPUrUUueEwgksDKN0cnE3iiXucuTaDdDK8uSPmk,6822
+mplang/edsl/serde.py,sha256=8K94laE8ObeGuBoF6m7g3A-xEe98EvqQ_6ZPPspddAY,11641
+mplang/edsl/tracer.py,sha256=g7P0WU6HvPYOhWMAflsX9YBLkAJyYI0YNPo9_P1nf-A,22528
+mplang/edsl/typing.py,sha256=Vp0r_oTyFrOhwwpVD_6XAX0sSsKvR04X-KdihkmD4QA,29287
+mplang/kernels/Makefile,sha256=5PoPpajcb_8ByPGNHzVytmovXUwkjJs_K8MbXX9qDYs,1033
+mplang/kernels/__init__.py,sha256=J_rDl9lAXd7QL3Nt_P3YX6j9yge7ssguSaHuafPZNKE,876
+mplang/kernels/gf128.cpp,sha256=WIvCr3MijzwJxMi1Wnfhm8aWT8oL0fia6FeyTmFJtPQ,5975
+mplang/kernels/ldpc.cpp,sha256=7n_FlI9btM3sSKdQOUC4f-PqTBAI98xccdaZIHX625M,2614
+mplang/kernels/okvs.cpp,sha256=7P_FkBY8LyEwnzupdlv48X8uKMjPUjXtzU2QlTjp-Q0,10311
+mplang/kernels/okvs_opt.cpp,sha256=qKkwHGGv7ayvIRW46PrQZZ7ej4b58Drbr3BlqEK506U,11707
+mplang/kernels/py_kernels.py,sha256=EXK2ByfRsVpBqp1etDyCb_qExT0obiytcNFXYpRyxpo,12070
+mplang/libs/collective.py,sha256=1RhwnpKwjLvbZ5UgmCMkO6brwqHEIdkW_zymRawwTW4,10668
+mplang/libs/device/__init__.py,sha256=UHPGQ9JVedXy6AQo-ixQooS2R8eTwJEu7-p8A5_po78,1248
+mplang/libs/device/api.py,sha256=lyMlNIFXQFuht_jaWc9uFkHRybv9OQC59nQ7USR5TWQ,28794
+mplang/libs/device/cluster.py,sha256=YUqYZ_IBS6rpV5ejUFP3kTxcTQHSyeDeuaJcsiFY_Js,12508
+mplang/libs/ml/__init__.py,sha256=kUR6V74kWmGnOUdmLuGAP1xebW7y-NBOmHyHK-kdqWg,784
+mplang/libs/ml/sgb.py,sha256=g5x1PQnV7oRzheyu6PiDooqmBWI9jMvLHkIHrNnMoDk,60211
+mplang/libs/mpc/__init__.py,sha256=0wTdCjWmoW4Ka4j8ItIfENd8t5uzgjFu0OhOtaeHlpY,1396
+mplang/libs/mpc/_utils.py,sha256=53mDJj_LTIJe6RWkR85-2nXha2KwYQqCK0PBG80MOl0,3209
+mplang/libs/mpc/analytics/__init__.py,sha256=8_1Sm05nrO2ISat1hNZT6UXHpKQ-SDBby1eeS-wm_fE,1204
+mplang/libs/mpc/analytics/aggregation.py,sha256=FXEp8Aa9NkS9cwQCqVxQBMa-fnw828QnyGdlM5IKb-4,11645
+mplang/libs/mpc/analytics/groupby.md,sha256=_3oCouzGKlWt3OBMIFkhN25gyzJj6DlGpq-B78_PSlU,4679
+mplang/libs/mpc/analytics/groupby.py,sha256=KjhZgB0EJwcShsIiE49HFzowgYVkYA7IQBx71_cj2m4,11312
+mplang/libs/mpc/analytics/permutation.py,sha256=FOXarjeKTZChf9dN7kaHhwQYytIQqRYMVsspovBKTnI,13811
+mplang/libs/mpc/common/constants.py,sha256=YRRE8WkLGfdOZzicc976hPsPeZZkqho3aW41i5Mj72s,1316
+mplang/libs/mpc/ot/__init__.py,sha256=9Ivef7QnAzoCBtIohh2kzOSdfslcNftOwKV_d9Qv4z0,951
+mplang/libs/mpc/ot/base.py,sha256=ydM6KRB4ytv5AVOZaCQ_GV54MsqmPhIE2JUUwoavi-c,7009
+mplang/libs/mpc/ot/extension.py,sha256=2Zk43JTpd7a0iorV7KWbyo_ebJi3G11QlxT9-8WJvBE,17452
+mplang/libs/mpc/ot/silent.py,sha256=lO8qhlejx0crIT_8PJcRU-lyUTgIsrE26hkoBzbR7sY,7867
+mplang/libs/mpc/psi/__init__.py,sha256=mpevlx3Z5_u9Q1McDZBBIGHApeO9julgiM09GToxxEA,1231
+mplang/libs/mpc/psi/cuckoo.py,sha256=BQE0CQ7OZWuz_E-ICJ55PDH6hG2T0McbR1QTLmf9HC0,7743
+mplang/libs/mpc/psi/okvs.py,sha256=bNUA_7LEkSw9Yw-v39G3qSfE4VtLHYj_a752WyTXXqc,1573
+mplang/libs/mpc/psi/okvs_gct.py,sha256=1XPw1CgQfuaspSRP7A2GbxyPJ66qW_sRnWmsWc_vG4w,3366
+mplang/libs/mpc/psi/oprf.py,sha256=JSSmK6JK9R8wKEVzmw_4BFa1wFA4UJqxrNK_ayBxE-Q,13763
+mplang/libs/mpc/psi/rr22.py,sha256=LXHqY_S3H7yEScGW6KO7NbliKIkB2prqOhnw1Vb2fE8,11188
+mplang/libs/mpc/psi/unbalanced.py,sha256=-0qAbYqw7vn_hBTb8E71z5RljO6-x8Tcn47TBH1v2_M,7421
+mplang/libs/mpc/vole/__init__.py,sha256=2dU4X6n73HoK-YCiCl4b36SkLRKR6rofe2xxLxBz6Rc,968
+mplang/libs/mpc/vole/gilboa.py,sha256=Lt9n9Ee_wgekxjvUdL0ed4eLpiQJ8EB4_YhOVqI-vBc,9942
+mplang/libs/mpc/vole/ldpc.py,sha256=gOmIbyOjkGE5lewyatl3p6FizNNH8LZ_1oOhp_-TOck,12780
+mplang/libs/mpc/vole/silver.py,sha256=EIxhpFIVNBemgeIZzCu5Cz_4wysxRm9b1Xfu0xiweVQ,12218
+mplang/runtime/__init__.py,sha256=VdUwJ3kDaI46FvGw7iMGwcsjt0HTGmmRmaBwj99xKIw,620
+mplang/runtime/dialect_state.py,sha256=HxO1i4kSOujS2tQzAF9-WmI3nChSaGgupf2_07dHetY,1277
+mplang/runtime/interpreter.py,sha256=jf79iH3JxQklzBPI2z1u8lnNo3CGJ0JBUiigPUQbevc,32256
+mplang/runtime/object_store.py,sha256=yT6jtKG2GUEJVmpq3gnQ8mCMvUFYzgBciC5A-J5KRdk,5998
+mplang/runtime/value.py,sha256=EqlhSgxLTJi_FF3ppyKjMe4eHS6-ROx-zK1YesG1U4o,4311
+mplang/utils/__init__.py,sha256=toubeyISiT6WDdITdfAvdY2iXVZU3PKVNWVeC9sYxuA,947
+mplang/utils/func_utils.py,sha256=aZ-X43w8JKJgiF-IUMS0G7QqrNeoTM5ZPzRNd-tKxpw,5180
+mplang_nightly-0.1.dev270.dist-info/METADATA,sha256=SJloPENuQIObfO0jrJ62IITYz3Hbl4y1Lo-mwN58-Yc,16783
+mplang_nightly-0.1.dev270.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+mplang_nightly-0.1.dev270.dist-info/entry_points.txt,sha256=mG1oJT-GAjQR834a62_QIWb7litzWPPyVnwFqm-rWuY,55
+mplang_nightly-0.1.dev270.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+mplang_nightly-0.1.dev270.dist-info/RECORD,,

mplang/v1/__init__.py

@@ -1,157 +0,0 @@
-# Copyright 2025 Ant Group Co., Ltd.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-"""Multi-Party Programming Language for Secure Computation."""
-
-# Version is managed by hatch-vcs and available after package installation
-try:
-    from importlib.metadata import PackageNotFoundError, version
-
-    __version__ = version("mplang")
-except PackageNotFoundError:
-    # Fallback for development/editable installs when package is not installed
-    __version__ = "0.0.0-dev"
-
-from mplang.v1 import analysis
-from mplang.v1._device import device, get_dev_attr, is_device_obj, put, set_dev_attr
-from mplang.v1.core import (
-    ClusterSpec,
-    Device,
-    DType,
-    InterpContext,
-    IrReader,
-    IrWriter,
-    Mask,
-    MPContext,
-    MPObject,
-    MPType,
-    Node,
-    Rank,
-    RuntimeInfo,
-    Shape,
-    TableType,
-    TensorType,
-    TraceContext,
-    TracedFunction,
-    cur_ctx,
-    function,
-    pconv,
-    peval,
-    pshfl,
-    pshfl_s,
-    set_ctx,
-    trace,
-    uniform_cond,
-    while_loop,
-    with_ctx,
-)
-from mplang.v1.host import CompileOptions, compile, evaluate, fetch
-from mplang.v1.runtime.driver import Driver
-from mplang.v1.runtime.simulation import Simulator
-from mplang.v1.simp.api import (
-    constant,
-    debug_print,
-    prand,
-    prank,
-    run,
-    run_at,
-    run_jax,
-    run_jax_at,
-    run_sql,
-    run_sql_at,
-    set_mask,
-)
-from mplang.v1.simp.mpi import allgather_m, bcast_m, gather_m, p2p, scatter_m
-from mplang.v1.simp.party import P0, P1, P2, P2P, P, Party, load_module
-from mplang.v1.simp.random import key_split, pperm, prandint, ukey, urandint
-from mplang.v1.simp.smpc import reveal, reveal_to, seal, seal_from, srun_jax
-
-# Public API
-__all__ = [
-    "P0",
-    "P1",
-    "P2",
-    "P2P",
-    "ClusterSpec",
-    "CompileOptions",
-    "DType",
-    "Device",
-    "Driver",
-    "InterpContext",
-    "IrReader",
-    "IrWriter",
-    "MPContext",
-    "MPObject",
-    "MPType",
-    "Mask",
-    "Node",
-    "P",
-    "Party",
-    "Rank",
-    "RuntimeInfo",
-    "Shape",
-    "Simulator",
-    "TableType",
-    "TensorType",
-    "TraceContext",
-    "TracedFunction",
-    "__version__",
-    "allgather_m",
-    "analysis",
-    "bcast_m",
-    "compile",
-    "constant",
-    "cur_ctx",
-    "debug_print",
-    "device",
-    "evaluate",
-    "fetch",
-    "function",
-    "gather_m",
-    "get_dev_attr",
-    "is_device_obj",
-    "key_split",
-    "load_module",
-    "p2p",
-    "pconv",
-    "peval",
-    "pperm",
-    "prand",
-    "prandint",
-    "prank",
-    "pshfl",
-    "pshfl_s",
-    "put",
-    "reveal",
-    "reveal_to",
-    "run",
-    "run_at",
-    "run_jax",
-    "run_jax_at",
-    "run_sql",
-    "run_sql_at",
-    "scatter_m",
-    "seal",
-    "seal_from",
-    "set_ctx",
-    "set_dev_attr",
-    "set_mask",
-    "srun_jax",
-    "trace",
-    "ukey",
-    "uniform_cond",
-    "urandint",
-    "while_loop",
-    "with_ctx",
-]