tapewasm 0.1.0

package/LICENSE

@@ -0,0 +1,201 @@
+                                 Apache License
+                           Version 2.0, January 2004
+                        http://www.apache.org/licenses/
+
+   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+   1. Definitions.
+
+      "License" shall mean the terms and conditions for use, reproduction,
+      and distribution as defined by Sections 1 through 9 of this document.
+
+      "Licensor" shall mean the copyright owner or entity authorized by
+      the copyright owner that is granting the License.
+
+      "Legal Entity" shall mean the union of the acting entity and all
+      other entities that control, are controlled by, or are under common
+      control with that entity. For the purposes of this definition,
+      "control" means (i) the power, direct or indirect, to cause the
+      direction or management of such entity, whether by contract or
+      otherwise, or (ii) ownership of fifty percent (50%) or more of the
+      outstanding shares, or (iii) beneficial ownership of such entity.
+
+      "You" (or "Your") shall mean an individual or Legal Entity
+      exercising permissions granted by this License.
+
+      "Source" form shall mean the preferred form for making modifications,
+      including but not limited to software source code, documentation
+      source, and configuration files.
+
+      "Object" form shall mean any form resulting from mechanical
+      transformation or translation of a Source form, including but
+      not limited to compiled object code, generated documentation,
+      and conversions to other media types.
+
+      "Work" shall mean the work of authorship, whether in Source or
+      Object form, made available under the License, as indicated by a
+      copyright notice that is included in or attached to the work
+      (an example is provided in the Appendix below).
+
+      "Derivative Works" shall mean any work, whether in Source or Object
+      form, that is based on (or derived from) the Work and for which the
+      editorial revisions, annotations, elaborations, or other modifications
+      represent, as a whole, an original work of authorship. For the purposes
+      of this License, Derivative Works shall not include works that remain
+      separable from, or merely link (or bind by name) to the interfaces of,
+      the Work and Derivative Works thereof.
+
+      "Contribution" shall mean any work of authorship, including
+      the original version of the Work and any modifications or additions
+      to that Work or Derivative Works thereof, that is intentionally
+      submitted to Licensor for inclusion in the Work by the copyright owner
+      or by an individual or Legal Entity authorized to submit on behalf of
+      the copyright owner. For the purposes of this definition, "submitted"
+      means any form of electronic, verbal, or written communication sent
+      to the Licensor or its representatives, including but not limited to
+      communication on electronic mailing lists, source code control systems,
+      and issue tracking systems that are managed by, or on behalf of, the
+      Licensor for the purpose of discussing and improving the Work, but
+      excluding communication that is conspicuously marked or otherwise
+      designated in writing by the copyright owner as "Not a Contribution."
+
+      "Contributor" shall mean Licensor and any individual or Legal Entity
+      on behalf of whom a Contribution has been received by Licensor and
+      subsequently incorporated within the Work.
+
+   2. Grant of Copyright License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      copyright license to reproduce, prepare Derivative Works of,
+      publicly display, publicly perform, sublicense, and distribute the
+      Work and such Derivative Works in Source or Object form.
+
+   3. Grant of Patent License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      (except as stated in this section) patent license to make, have made,
+      use, offer to sell, sell, import, and otherwise transfer the Work,
+      where such license applies only to those patent claims licensable
+      by such Contributor that are necessarily infringed by their
+      Contribution(s) alone or by combination of their Contribution(s)
+      with the Work to which such Contribution(s) was submitted. If You
+      institute patent litigation against any entity (including a
+      cross-claim or counterclaim in a lawsuit) alleging that the Work
+      or a Contribution incorporated within the Work constitutes direct
+      or contributory patent infringement, then any patent licenses
+      granted to You under this License for that Work shall terminate
+      as of the date such litigation is filed.
+
+   4. Redistribution. You may reproduce and distribute copies of the
+      Work or Derivative Works thereof in any medium, with or without
+      modifications, and in Source or Object form, provided that You
+      meet the following conditions:
+
+      (a) You must give any other recipients of the Work or
+          Derivative Works a copy of this License; and
+
+      (b) You must cause any modified files to carry prominent notices
+          stating that You changed the files; and
+
+      (c) You must retain, in the Source form of any Derivative Works
+          that You distribute, all copyright, patent, trademark, and
+          attribution notices from the Source form of the Work,
+          excluding those notices that do not pertain to any part of
+          the Derivative Works; and
+
+      (d) If the Work includes a "NOTICE" text file as part of its
+          distribution, then any Derivative Works that You distribute must
+          include a readable copy of the attribution notices contained
+          within such NOTICE file, excluding those notices that do not
+          pertain to any part of the Derivative Works, in at least one
+          of the following places: within a NOTICE text file distributed
+          as part of the Derivative Works; within the Source form or
+          documentation, if provided along with the Derivative Works; or,
+          within a display generated by the Derivative Works, if and
+          wherever such third-party notices normally appear. The contents
+          of the NOTICE file are for informational purposes only and
+          do not modify the License. You may add Your own attribution
+          notices within Derivative Works that You distribute, alongside
+          or as an addendum to the NOTICE text from the Work, provided
+          that such additional attribution notices cannot be construed
+          as modifying the License.
+
+      You may add Your own copyright statement to Your modifications and
+      may provide additional or different license terms and conditions
+      for use, reproduction, or distribution of Your modifications, or
+      for any such Derivative Works as a whole, provided Your use,
+      reproduction, and distribution of the Work otherwise complies with
+      the conditions stated in this License.
+
+   5. Submission of Contributions. Unless You explicitly state otherwise,
+      any Contribution intentionally submitted for inclusion in the Work
+      by You to the Licensor shall be under the terms and conditions of
+      this License, without any additional terms or conditions.
+      Notwithstanding the above, nothing herein shall supersede or modify
+      the terms of any separate license agreement you may have executed
+      with Licensor regarding such Contributions.
+
+   6. Trademarks. This License does not grant permission to use the trade
+      names, trademarks, service marks, or product names of the Licensor,
+      except as required for describing the origin of the Work and
+      reproducing the content of the NOTICE file.
+
+   7. Disclaimer of Warranty. Unless required by applicable law or
+      agreed to in writing, Licensor provides the Work (and each
+      Contributor provides its Contributions) on an "AS IS" BASIS,
+      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+      implied, including, without limitation, any warranties or conditions
+      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+      PARTICULAR PURPOSE. You are solely responsible for determining the
+      appropriateness of using or redistributing the Work and assume any
+      risks associated with Your exercise of permissions under this License.
+
+   8. Limitation of Liability. In no event and under no legal theory,
+      whether in tort (including negligence), contract, or otherwise,
+      unless required by applicable law (such as deliberate and grossly
+      negligent acts) or agreed to in writing, shall any Contributor be
+      liable to You for damages, including any direct, indirect, special,
+      incidental, or consequential damages of any character arising as a
+      result of this License or out of the use or inability to use the
+      Work (including but not limited to damages for loss of goodwill,
+      work stoppage, computer failure or malfunction, or any and all
+      other commercial damages or losses), even if such Contributor
+      has been advised of the possibility of such damages.
+
+   9. Accepting Warranty or Support. While redistributing
+      the Work or Derivative Works thereof, You may choose to offer,
+      and charge a fee for, acceptance of support, warranty, indemnity,
+      or other liability obligations and/or rights consistent with this
+      License. However, in accepting such obligations, You may act only
+      on Your own behalf and on Your sole responsibility, not on behalf
+      of any other Contributor, and only if You agree to indemnify,
+      defend, and hold each Contributor harmless for any liability
+      incurred by, or claims asserted against, such Contributor by reason
+      of your accepting any such warranty or support.
+
+   END OF TERMS AND CONDITIONS
+
+   APPENDIX: How to apply the Apache License to your work.
+
+      To apply the Apache License to your work, attach the following
+      boilerplate notice, with the fields enclosed by brackets "[]"
+      replaced with your own identifying information. (Don't include
+      the brackets!)  The text should be enclosed in the appropriate
+      comment syntax for the file format. We also recommend that a
+      file or class name and description of purpose be included on the
+      same "printed page" as the copyright notice for easier
+      identification within third-party archives.
+
+   Copyright 2026 habakan
+
+   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.

package/README.md

@@ -0,0 +1,10 @@
+# tapewasm
+
+The wasm-bindgen API: compile a tape to a WebAssembly module, then sample it
+with [nuts-rs](https://github.com/pymc-devs/nuts-rs).
+
+See the [repository](https://github.com/habakan/tapewasm) for what this is for
+and how a front end uses it. The npm package of the same name is this crate
+built with `wasm-pack`.
+
+Licensed under Apache-2.0.

package/index.d.ts

@@ -0,0 +1,11 @@
+// Types for `index.js`; the shapes come from `pkg/`, which `make wasm` writes.
+export {
+  AotSampler,
+  CompiledTape,
+  compileTape,
+  tapewasmVersion,
+  setAotExports,
+  clearAotExports,
+  sharedMemory,
+} from "./pkg/tapewasm.js";
+export { default } from "./pkg/tapewasm.js";

package/index.js

@@ -0,0 +1,15 @@
+// Public facade: re-exports the wasm-bindgen bindings so callers never reach
+// into `pkg/`. Plain `.js` because a package entry point has to load without
+// `--experimental-strip-types`, and there is no TypeScript syntax here to lose.
+
+import init from "./pkg/tapewasm.js";
+export {
+  AotSampler,
+  CompiledTape,
+  compileTape,
+  tapewasmVersion,
+  setAotExports,
+  clearAotExports,
+  sharedMemory,
+} from "./pkg/tapewasm.js";
+export default init;

package/package.json

@@ -0,0 +1,27 @@
+{
+  "name": "tapewasm",
+  "version": "0.1.0",
+  "description": "Compile an autodiff tape to a WebAssembly module and sample it with nuts-rs, in the browser.",
+  "type": "module",
+  "main": "./index.js",
+  "module": "./index.js",
+  "types": "./index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./index.d.ts",
+      "default": "./index.js"
+    },
+    "./pkg/*": "./pkg/*"
+  },
+  "files": ["index.js", "index.d.ts", "pkg/", "LICENSE", "README.md"],
+  "scripts": {
+    "build:wasm": "make -C .. wasm"
+  },
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/habakan/tapewasm.git"
+  },
+  "homepage": "https://github.com/habakan/tapewasm#readme",
+  "keywords": ["autodiff", "gradient", "bayesian", "mcmc", "nuts", "webassembly", "wasm"],
+  "license": "Apache-2.0"
+}

package/pkg/LICENSE

@@ -0,0 +1,201 @@
+                                 Apache License
+                           Version 2.0, January 2004
+                        http://www.apache.org/licenses/
+
+   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+   1. Definitions.
+
+      "License" shall mean the terms and conditions for use, reproduction,
+      and distribution as defined by Sections 1 through 9 of this document.
+
+      "Licensor" shall mean the copyright owner or entity authorized by
+      the copyright owner that is granting the License.
+
+      "Legal Entity" shall mean the union of the acting entity and all
+      other entities that control, are controlled by, or are under common
+      control with that entity. For the purposes of this definition,
+      "control" means (i) the power, direct or indirect, to cause the
+      direction or management of such entity, whether by contract or
+      otherwise, or (ii) ownership of fifty percent (50%) or more of the
+      outstanding shares, or (iii) beneficial ownership of such entity.
+
+      "You" (or "Your") shall mean an individual or Legal Entity
+      exercising permissions granted by this License.
+
+      "Source" form shall mean the preferred form for making modifications,
+      including but not limited to software source code, documentation
+      source, and configuration files.
+
+      "Object" form shall mean any form resulting from mechanical
+      transformation or translation of a Source form, including but
+      not limited to compiled object code, generated documentation,
+      and conversions to other media types.
+
+      "Work" shall mean the work of authorship, whether in Source or
+      Object form, made available under the License, as indicated by a
+      copyright notice that is included in or attached to the work
+      (an example is provided in the Appendix below).
+
+      "Derivative Works" shall mean any work, whether in Source or Object
+      form, that is based on (or derived from) the Work and for which the
+      editorial revisions, annotations, elaborations, or other modifications
+      represent, as a whole, an original work of authorship. For the purposes
+      of this License, Derivative Works shall not include works that remain
+      separable from, or merely link (or bind by name) to the interfaces of,
+      the Work and Derivative Works thereof.
+
+      "Contribution" shall mean any work of authorship, including
+      the original version of the Work and any modifications or additions
+      to that Work or Derivative Works thereof, that is intentionally
+      submitted to Licensor for inclusion in the Work by the copyright owner
+      or by an individual or Legal Entity authorized to submit on behalf of
+      the copyright owner. For the purposes of this definition, "submitted"
+      means any form of electronic, verbal, or written communication sent
+      to the Licensor or its representatives, including but not limited to
+      communication on electronic mailing lists, source code control systems,
+      and issue tracking systems that are managed by, or on behalf of, the
+      Licensor for the purpose of discussing and improving the Work, but
+      excluding communication that is conspicuously marked or otherwise
+      designated in writing by the copyright owner as "Not a Contribution."
+
+      "Contributor" shall mean Licensor and any individual or Legal Entity
+      on behalf of whom a Contribution has been received by Licensor and
+      subsequently incorporated within the Work.
+
+   2. Grant of Copyright License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      copyright license to reproduce, prepare Derivative Works of,
+      publicly display, publicly perform, sublicense, and distribute the
+      Work and such Derivative Works in Source or Object form.
+
+   3. Grant of Patent License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      (except as stated in this section) patent license to make, have made,
+      use, offer to sell, sell, import, and otherwise transfer the Work,
+      where such license applies only to those patent claims licensable
+      by such Contributor that are necessarily infringed by their
+      Contribution(s) alone or by combination of their Contribution(s)
+      with the Work to which such Contribution(s) was submitted. If You
+      institute patent litigation against any entity (including a
+      cross-claim or counterclaim in a lawsuit) alleging that the Work
+      or a Contribution incorporated within the Work constitutes direct
+      or contributory patent infringement, then any patent licenses
+      granted to You under this License for that Work shall terminate
+      as of the date such litigation is filed.
+
+   4. Redistribution. You may reproduce and distribute copies of the
+      Work or Derivative Works thereof in any medium, with or without
+      modifications, and in Source or Object form, provided that You
+      meet the following conditions:
+
+      (a) You must give any other recipients of the Work or
+          Derivative Works a copy of this License; and
+
+      (b) You must cause any modified files to carry prominent notices
+          stating that You changed the files; and
+
+      (c) You must retain, in the Source form of any Derivative Works
+          that You distribute, all copyright, patent, trademark, and
+          attribution notices from the Source form of the Work,
+          excluding those notices that do not pertain to any part of
+          the Derivative Works; and
+
+      (d) If the Work includes a "NOTICE" text file as part of its
+          distribution, then any Derivative Works that You distribute must
+          include a readable copy of the attribution notices contained
+          within such NOTICE file, excluding those notices that do not
+          pertain to any part of the Derivative Works, in at least one
+          of the following places: within a NOTICE text file distributed
+          as part of the Derivative Works; within the Source form or
+          documentation, if provided along with the Derivative Works; or,
+          within a display generated by the Derivative Works, if and
+          wherever such third-party notices normally appear. The contents
+          of the NOTICE file are for informational purposes only and
+          do not modify the License. You may add Your own attribution
+          notices within Derivative Works that You distribute, alongside
+          or as an addendum to the NOTICE text from the Work, provided
+          that such additional attribution notices cannot be construed
+          as modifying the License.
+
+      You may add Your own copyright statement to Your modifications and
+      may provide additional or different license terms and conditions
+      for use, reproduction, or distribution of Your modifications, or
+      for any such Derivative Works as a whole, provided Your use,
+      reproduction, and distribution of the Work otherwise complies with
+      the conditions stated in this License.
+
+   5. Submission of Contributions. Unless You explicitly state otherwise,
+      any Contribution intentionally submitted for inclusion in the Work
+      by You to the Licensor shall be under the terms and conditions of
+      this License, without any additional terms or conditions.
+      Notwithstanding the above, nothing herein shall supersede or modify
+      the terms of any separate license agreement you may have executed
+      with Licensor regarding such Contributions.
+
+   6. Trademarks. This License does not grant permission to use the trade
+      names, trademarks, service marks, or product names of the Licensor,
+      except as required for describing the origin of the Work and
+      reproducing the content of the NOTICE file.
+
+   7. Disclaimer of Warranty. Unless required by applicable law or
+      agreed to in writing, Licensor provides the Work (and each
+      Contributor provides its Contributions) on an "AS IS" BASIS,
+      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+      implied, including, without limitation, any warranties or conditions
+      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+      PARTICULAR PURPOSE. You are solely responsible for determining the
+      appropriateness of using or redistributing the Work and assume any
+      risks associated with Your exercise of permissions under this License.
+
+   8. Limitation of Liability. In no event and under no legal theory,
+      whether in tort (including negligence), contract, or otherwise,
+      unless required by applicable law (such as deliberate and grossly
+      negligent acts) or agreed to in writing, shall any Contributor be
+      liable to You for damages, including any direct, indirect, special,
+      incidental, or consequential damages of any character arising as a
+      result of this License or out of the use or inability to use the
+      Work (including but not limited to damages for loss of goodwill,
+      work stoppage, computer failure or malfunction, or any and all
+      other commercial damages or losses), even if such Contributor
+      has been advised of the possibility of such damages.
+
+   9. Accepting Warranty or Support. While redistributing
+      the Work or Derivative Works thereof, You may choose to offer,
+      and charge a fee for, acceptance of support, warranty, indemnity,
+      or other liability obligations and/or rights consistent with this
+      License. However, in accepting such obligations, You may act only
+      on Your own behalf and on Your sole responsibility, not on behalf
+      of any other Contributor, and only if You agree to indemnify,
+      defend, and hold each Contributor harmless for any liability
+      incurred by, or claims asserted against, such Contributor by reason
+      of your accepting any such warranty or support.
+
+   END OF TERMS AND CONDITIONS
+
+   APPENDIX: How to apply the Apache License to your work.
+
+      To apply the Apache License to your work, attach the following
+      boilerplate notice, with the fields enclosed by brackets "[]"
+      replaced with your own identifying information. (Don't include
+      the brackets!)  The text should be enclosed in the appropriate
+      comment syntax for the file format. We also recommend that a
+      file or class name and description of purpose be included on the
+      same "printed page" as the copyright notice for easier
+      identification within third-party archives.
+
+   Copyright 2026 habakan
+
+   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.

package/pkg/README.md

@@ -0,0 +1,10 @@
+# tapewasm
+
+The wasm-bindgen API: compile a tape to a WebAssembly module, then sample it
+with [nuts-rs](https://github.com/pymc-devs/nuts-rs).
+
+See the [repository](https://github.com/habakan/tapewasm) for what this is for
+and how a front end uses it. The npm package of the same name is this crate
+built with `wasm-pack`.
+
+Licensed under Apache-2.0.

package/pkg/package.json

@@ -0,0 +1,28 @@
+{
+  "name": "tapewasm",
+  "type": "module",
+  "description": "Compile an autodiff tape to a wasm module and sample it with nuts-rs, in the browser. The wasm-bindgen API over tapewasm-codegen.",
+  "version": "0.1.0",
+  "license": "Apache-2.0",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/habakan/tapewasm"
+  },
+  "files": [
+    "tapewasm_bg.wasm",
+    "tapewasm.js",
+    "tapewasm.d.ts"
+  ],
+  "main": "tapewasm.js",
+  "types": "tapewasm.d.ts",
+  "sideEffects": [
+    "./snippets/*"
+  ],
+  "keywords": [
+    "autodiff",
+    "wasm",
+    "bayesian",
+    "mcmc",
+    "nuts"
+  ]
+}

package/pkg/snippets/tapewasm-86b4af57c39247cc/js/aot_bridge.js

@@ -0,0 +1,41 @@
+// Bridge between tapewasm.wasm and a per-model compiled wasm, which imports
+// tapewasm's memory and exports `log_prob_grad`, `tapewasm_layout_id` and
+// `tapewasm_abi_version`.
+//
+// The binding is per page while the scratch buffer belongs to one model, so
+// the sampler reads the id back to refuse a mismatched pair.
+
+let aotLogProbGrad = null;
+// NaN means nothing is bound, or no id is exported; no u32 id collides with it.
+let aotLayoutId = NaN;
+// Likewise: a module from before the global existed reads as unknown, not as 0.
+let aotAbiVersion = NaN;
+
+export function set_aot_exports(exports) {
+  aotLogProbGrad = exports.log_prob_grad;
+  const g = exports.tapewasm_layout_id;
+  aotLayoutId = g ? g.value >>> 0 : NaN;
+  const v = exports.tapewasm_abi_version;
+  aotAbiVersion = v ? v.value >>> 0 : NaN;
+}
+
+export function clear_aot_exports() {
+  aotLogProbGrad = null;
+  aotLayoutId = NaN;
+  aotAbiVersion = NaN;
+}
+
+export function aot_layout_id() {
+  return aotLayoutId;
+}
+
+export function aot_abi_version() {
+  return aotAbiVersion;
+}
+
+export function aot_logp(paramsPtr, gradsPtr, nParams, scratchPtr) {
+  if (!aotLogProbGrad) {
+    throw new Error("no module bound — call setAotExports() before sampling");
+  }
+  return aotLogProbGrad(paramsPtr, gradsPtr, nParams, scratchPtr);
+}

package/pkg/tapewasm.d.ts

@@ -0,0 +1,146 @@
+/* tslint:disable */
+/* eslint-disable */
+
+/**
+ * A sampler over one compiled module.
+ *
+ * It reaches the module alone: nothing that recorded or emitted the tape is
+ * on this path, so a page that loads a module built earlier needs no
+ * compiler in its bundle. Bind the module with `setAotExports` first.
+ */
+export class AotSampler {
+    free(): void;
+    [Symbol.dispose](): void;
+    /**
+     * `[log_prob, d/dparam...]`.
+     */
+    logProbGrad(params: Float64Array): Float64Array;
+    /**
+     * `scratch_init` is the buffer the module works in — zeroed primals and
+     * adjoints followed by the re-rolled loops' constant table, which is what
+     * `aotScratchInit` returns and what a compiler should record beside the
+     * module. `layout_id` is the module's `tapewasm_layout_id` global, checked
+     * against the bound exports before each run so a module and a scratch
+     * buffer built for different models cannot be used together.
+     *
+     * `param_names` only names a parameter in a rejected starting point; pass
+     * an empty array to go without.
+     */
+    constructor(n_params: number, scratch_init: Float64Array, layout_id: number, param_names: string[]);
+    /**
+     * `num_warmup + num_draws` draws, row-major, `n_params` wide.
+     */
+    sample(init: Float64Array, num_warmup: number, num_draws: number, seed: bigint): Float64Array;
+    readonly nParams: number;
+}
+
+/**
+ * What [`compile_tape`] produced, and everything [`AotSampler`] needs from it.
+ */
+export class CompiledTape {
+    private constructor();
+    free(): void;
+    [Symbol.dispose](): void;
+    readonly layoutId: number;
+    readonly nParams: number;
+    readonly scratchInit: Float64Array;
+    /**
+     * The module. Instantiate it against [`shared_memory`] and the `Math`
+     * imports, then bind it with `setAotExports`.
+     */
+    readonly wasm: Uint8Array;
+}
+
+/**
+ * Release the bound module's exports. The next draw will throw.
+ */
+export function clearAotExports(): void;
+
+/**
+ * Compile a tape written by another front end.
+ *
+ * `tape` is the text format `tapewasm_codegen::tape_text` documents: one
+ * instruction per line, operands naming instructions rather than nodes. A
+ * front end that can build a tape reaches the emitter this way — including
+ * one that is not Rust and not in this module.
+ *
+ * The format is not an artifact and carries no compatibility promise: a tape
+ * is written and consumed inside one call.
+ */
+export function compileTape(tape: string): CompiledTape;
+
+/**
+ * Forwards Rust panics to `console.error` with a message and backtrace rather
+ * than an opaque `RuntimeError: unreachable`. Diagnostics: the instance still traps.
+ */
+export function init_panic_hook(): void;
+
+/**
+ * Bind a freshly-instantiated model module's exports so sampling dispatches
+ * through it. Pass `instance.exports`.
+ *
+ * One binding serves the whole page, so a page holding several models has to
+ * re-bind before sampling a different one. The sampler compares the bound
+ * module's `tapewasm_layout_id` against the model it belongs to and refuses
+ * the pair rather than running it against the wrong scratch buffer.
+ */
+export function setAotExports(exports: any): void;
+
+/**
+ * The linear memory backing this module. Pass as the `tapewasm.memory`
+ * import when instantiating a model module so the two share buffers.
+ */
+export function sharedMemory(): any;
+
+export function tapewasmVersion(): string;
+
+export type InitInput = RequestInfo | URL | Response | BufferSource | WebAssembly.Module;
+
+export interface InitOutput {
+    readonly memory: WebAssembly.Memory;
+    readonly __wbg_aotsampler_free: (a: number, b: number) => void;
+    readonly __wbg_compiledtape_free: (a: number, b: number) => void;
+    readonly aotsampler_logProbGrad: (a: number, b: number, c: number) => [number, number, number, number];
+    readonly aotsampler_nParams: (a: number) => number;
+    readonly aotsampler_new: (a: number, b: number, c: number, d: number, e: number, f: number) => [number, number, number];
+    readonly aotsampler_sample: (a: number, b: number, c: number, d: number, e: number, f: bigint) => [number, number, number, number];
+    readonly clearAotExports: () => void;
+    readonly compileTape: (a: number, b: number) => [number, number, number];
+    readonly compiledtape_layoutId: (a: number) => number;
+    readonly compiledtape_nParams: (a: number) => number;
+    readonly compiledtape_scratchInit: (a: number) => [number, number];
+    readonly compiledtape_wasm: (a: number) => [number, number];
+    readonly init_panic_hook: () => void;
+    readonly setAotExports: (a: any) => void;
+    readonly sharedMemory: () => any;
+    readonly tapewasmVersion: () => [number, number];
+    readonly __wbindgen_malloc: (a: number, b: number) => number;
+    readonly __wbindgen_realloc: (a: number, b: number, c: number, d: number) => number;
+    readonly __wbindgen_free: (a: number, b: number, c: number) => void;
+    readonly __wbindgen_externrefs: WebAssembly.Table;
+    readonly __externref_table_dealloc: (a: number) => void;
+    readonly __externref_table_alloc: () => number;
+    readonly __wbindgen_start: () => void;
+}
+
+export type SyncInitInput = BufferSource | WebAssembly.Module;
+
+/**
+ * Instantiates the given `module`, which can either be bytes or
+ * a precompiled `WebAssembly.Module`.
+ *
+ * @param {{ module: SyncInitInput }} module - Passing `SyncInitInput` directly is deprecated.
+ *
+ * @returns {InitOutput}
+ */
+export function initSync(module: { module: SyncInitInput } | SyncInitInput): InitOutput;
+
+/**
+ * If `module_or_path` is {RequestInfo} or {URL}, makes a request and
+ * for everything else, calls `WebAssembly.instantiate` directly.
+ *
+ * @param {{ module_or_path: InitInput | Promise<InitInput> }} module_or_path - Passing `InitInput` directly is deprecated.
+ *
+ * @returns {Promise<InitOutput>}
+ */
+export default function __wbg_init (module_or_path?: { module_or_path: InitInput | Promise<InitInput> } | InitInput | Promise<InitInput>): Promise<InitOutput>;

package/pkg/tapewasm.js

@@ -0,0 +1,543 @@
+/* @ts-self-types="./tapewasm.d.ts" */
+import { aot_logp } from './snippets/tapewasm-86b4af57c39247cc/js/aot_bridge.js';
+import * as import1 from "./snippets/tapewasm-86b4af57c39247cc/js/aot_bridge.js"
+import * as import2 from "./snippets/tapewasm-86b4af57c39247cc/js/aot_bridge.js"
+import * as import3 from "./snippets/tapewasm-86b4af57c39247cc/js/aot_bridge.js"
+import * as import4 from "./snippets/tapewasm-86b4af57c39247cc/js/aot_bridge.js"
+
+
+/**
+ * A sampler over one compiled module.
+ *
+ * It reaches the module alone: nothing that recorded or emitted the tape is
+ * on this path, so a page that loads a module built earlier needs no
+ * compiler in its bundle. Bind the module with `setAotExports` first.
+ */
+export class AotSampler {
+    __destroy_into_raw() {
+        const ptr = this.__wbg_ptr;
+        this.__wbg_ptr = 0;
+        AotSamplerFinalization.unregister(this);
+        return ptr;
+    }
+    free() {
+        const ptr = this.__destroy_into_raw();
+        wasm.__wbg_aotsampler_free(ptr, 0);
+    }
+    /**
+     * `[log_prob, d/dparam...]`.
+     * @param {Float64Array} params
+     * @returns {Float64Array}
+     */
+    logProbGrad(params) {
+        const ptr0 = passArrayF64ToWasm0(params, wasm.__wbindgen_malloc);
+        const len0 = WASM_VECTOR_LEN;
+        const ret = wasm.aotsampler_logProbGrad(this.__wbg_ptr, ptr0, len0);
+        if (ret[3]) {
+            throw takeFromExternrefTable0(ret[2]);
+        }
+        var v2 = getArrayF64FromWasm0(ret[0], ret[1]).slice();
+        wasm.__wbindgen_free(ret[0], ret[1] * 8, 8);
+        return v2;
+    }
+    /**
+     * @returns {number}
+     */
+    get nParams() {
+        const ret = wasm.aotsampler_nParams(this.__wbg_ptr);
+        return ret >>> 0;
+    }
+    /**
+     * `scratch_init` is the buffer the module works in — zeroed primals and
+     * adjoints followed by the re-rolled loops' constant table, which is what
+     * `aotScratchInit` returns and what a compiler should record beside the
+     * module. `layout_id` is the module's `tapewasm_layout_id` global, checked
+     * against the bound exports before each run so a module and a scratch
+     * buffer built for different models cannot be used together.
+     *
+     * `param_names` only names a parameter in a rejected starting point; pass
+     * an empty array to go without.
+     * @param {number} n_params
+     * @param {Float64Array} scratch_init
+     * @param {number} layout_id
+     * @param {string[]} param_names
+     */
+    constructor(n_params, scratch_init, layout_id, param_names) {
+        const ptr0 = passArrayF64ToWasm0(scratch_init, wasm.__wbindgen_malloc);
+        const len0 = WASM_VECTOR_LEN;
+        const ptr1 = passArrayJsValueToWasm0(param_names, wasm.__wbindgen_malloc);
+        const len1 = WASM_VECTOR_LEN;
+        const ret = wasm.aotsampler_new(n_params, ptr0, len0, layout_id, ptr1, len1);
+        if (ret[2]) {
+            throw takeFromExternrefTable0(ret[1]);
+        }
+        this.__wbg_ptr = ret[0];
+        AotSamplerFinalization.register(this, this.__wbg_ptr, this);
+        return this;
+    }
+    /**
+     * `num_warmup + num_draws` draws, row-major, `n_params` wide.
+     * @param {Float64Array} init
+     * @param {number} num_warmup
+     * @param {number} num_draws
+     * @param {bigint} seed
+     * @returns {Float64Array}
+     */
+    sample(init, num_warmup, num_draws, seed) {
+        const ptr0 = passArrayF64ToWasm0(init, wasm.__wbindgen_malloc);
+        const len0 = WASM_VECTOR_LEN;
+        const ret = wasm.aotsampler_sample(this.__wbg_ptr, ptr0, len0, num_warmup, num_draws, seed);
+        if (ret[3]) {
+            throw takeFromExternrefTable0(ret[2]);
+        }
+        var v2 = getArrayF64FromWasm0(ret[0], ret[1]).slice();
+        wasm.__wbindgen_free(ret[0], ret[1] * 8, 8);
+        return v2;
+    }
+}
+if (Symbol.dispose) AotSampler.prototype[Symbol.dispose] = AotSampler.prototype.free;
+
+/**
+ * What [`compile_tape`] produced, and everything [`AotSampler`] needs from it.
+ */
+export class CompiledTape {
+    static __wrap(ptr) {
+        const obj = Object.create(CompiledTape.prototype);
+        obj.__wbg_ptr = ptr;
+        CompiledTapeFinalization.register(obj, obj.__wbg_ptr, obj);
+        return obj;
+    }
+    __destroy_into_raw() {
+        const ptr = this.__wbg_ptr;
+        this.__wbg_ptr = 0;
+        CompiledTapeFinalization.unregister(this);
+        return ptr;
+    }
+    free() {
+        const ptr = this.__destroy_into_raw();
+        wasm.__wbg_compiledtape_free(ptr, 0);
+    }
+    /**
+     * @returns {number}
+     */
+    get layoutId() {
+        const ret = wasm.compiledtape_layoutId(this.__wbg_ptr);
+        return ret >>> 0;
+    }
+    /**
+     * @returns {number}
+     */
+    get nParams() {
+        const ret = wasm.compiledtape_nParams(this.__wbg_ptr);
+        return ret >>> 0;
+    }
+    /**
+     * @returns {Float64Array}
+     */
+    get scratchInit() {
+        const ret = wasm.compiledtape_scratchInit(this.__wbg_ptr);
+        var v1 = getArrayF64FromWasm0(ret[0], ret[1]).slice();
+        wasm.__wbindgen_free(ret[0], ret[1] * 8, 8);
+        return v1;
+    }
+    /**
+     * The module. Instantiate it against [`shared_memory`] and the `Math`
+     * imports, then bind it with `setAotExports`.
+     * @returns {Uint8Array}
+     */
+    get wasm() {
+        const ret = wasm.compiledtape_wasm(this.__wbg_ptr);
+        var v1 = getArrayU8FromWasm0(ret[0], ret[1]).slice();
+        wasm.__wbindgen_free(ret[0], ret[1] * 1, 1);
+        return v1;
+    }
+}
+if (Symbol.dispose) CompiledTape.prototype[Symbol.dispose] = CompiledTape.prototype.free;
+
+/**
+ * Release the bound module's exports. The next draw will throw.
+ */
+export function clearAotExports() {
+    wasm.clearAotExports();
+}
+
+/**
+ * Compile a tape written by another front end.
+ *
+ * `tape` is the text format `tapewasm_codegen::tape_text` documents: one
+ * instruction per line, operands naming instructions rather than nodes. A
+ * front end that can build a tape reaches the emitter this way — including
+ * one that is not Rust and not in this module.
+ *
+ * The format is not an artifact and carries no compatibility promise: a tape
+ * is written and consumed inside one call.
+ * @param {string} tape
+ * @returns {CompiledTape}
+ */
+export function compileTape(tape) {
+    const ptr0 = passStringToWasm0(tape, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
+    const len0 = WASM_VECTOR_LEN;
+    const ret = wasm.compileTape(ptr0, len0);
+    if (ret[2]) {
+        throw takeFromExternrefTable0(ret[1]);
+    }
+    return CompiledTape.__wrap(ret[0]);
+}
+
+/**
+ * Forwards Rust panics to `console.error` with a message and backtrace rather
+ * than an opaque `RuntimeError: unreachable`. Diagnostics: the instance still traps.
+ */
+export function init_panic_hook() {
+    wasm.init_panic_hook();
+}
+
+/**
+ * Bind a freshly-instantiated model module's exports so sampling dispatches
+ * through it. Pass `instance.exports`.
+ *
+ * One binding serves the whole page, so a page holding several models has to
+ * re-bind before sampling a different one. The sampler compares the bound
+ * module's `tapewasm_layout_id` against the model it belongs to and refuses
+ * the pair rather than running it against the wrong scratch buffer.
+ * @param {any} exports
+ */
+export function setAotExports(exports) {
+    wasm.setAotExports(exports);
+}
+
+/**
+ * The linear memory backing this module. Pass as the `tapewasm.memory`
+ * import when instantiating a model module so the two share buffers.
+ * @returns {any}
+ */
+export function sharedMemory() {
+    const ret = wasm.sharedMemory();
+    return ret;
+}
+
+/**
+ * @returns {string}
+ */
+export function tapewasmVersion() {
+    let deferred1_0;
+    let deferred1_1;
+    try {
+        const ret = wasm.tapewasmVersion();
+        deferred1_0 = ret[0];
+        deferred1_1 = ret[1];
+        return getStringFromWasm0(ret[0], ret[1]);
+    } finally {
+        wasm.__wbindgen_free(deferred1_0, deferred1_1, 1);
+    }
+}
+function __wbg_get_imports() {
+    const import0 = {
+        __proto__: null,
+        __wbg_Error_67e7344beaa85059: function(arg0, arg1) {
+            const ret = Error(getStringFromWasm0(arg0, arg1));
+            return ret;
+        },
+        __wbg___wbindgen_memory_3f8442e22540244f: function() {
+            const ret = wasm.memory;
+            return ret;
+        },
+        __wbg___wbindgen_string_get_92ab86bb19cbc12f: function(arg0, arg1) {
+            const obj = arg1;
+            const ret = typeof(obj) === 'string' ? obj : undefined;
+            var ptr1 = isLikeNone(ret) ? 0 : passStringToWasm0(ret, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
+            var len1 = WASM_VECTOR_LEN;
+            getDataViewMemory0().setInt32(arg0 + 4 * 1, len1, true);
+            getDataViewMemory0().setInt32(arg0 + 4 * 0, ptr1, true);
+        },
+        __wbg___wbindgen_throw_5d9e815e6fdf150f: function(arg0, arg1) {
+            throw new Error(getStringFromWasm0(arg0, arg1));
+        },
+        __wbg_aot_logp_50f9146d22ee7ef4: function(arg0, arg1, arg2, arg3) {
+            const ret = aot_logp(arg0 >>> 0, arg1 >>> 0, arg2 >>> 0, arg3 >>> 0);
+            return ret;
+        },
+        __wbg_error_757e9472f8410341: function(arg0, arg1) {
+            let deferred0_0;
+            let deferred0_1;
+            try {
+                deferred0_0 = arg0;
+                deferred0_1 = arg1;
+                console.error(getStringFromWasm0(arg0, arg1));
+            } finally {
+                wasm.__wbindgen_free(deferred0_0, deferred0_1, 1);
+            }
+        },
+        __wbg_new_227d7c05414eb861: function() {
+            const ret = new Error();
+            return ret;
+        },
+        __wbg_stack_3b0d974bbf31e44f: function(arg0, arg1) {
+            const ret = arg1.stack;
+            const ptr1 = passStringToWasm0(ret, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
+            const len1 = WASM_VECTOR_LEN;
+            getDataViewMemory0().setInt32(arg0 + 4 * 1, len1, true);
+            getDataViewMemory0().setInt32(arg0 + 4 * 0, ptr1, true);
+        },
+        __wbindgen_init_externref_table: function() {
+            const table = wasm.__wbindgen_externrefs;
+            const offset = table.grow(4);
+            table.set(0, undefined);
+            table.set(offset + 0, undefined);
+            table.set(offset + 1, null);
+            table.set(offset + 2, true);
+            table.set(offset + 3, false);
+        },
+    };
+    return {
+        __proto__: null,
+        "./tapewasm_bg.js": import0,
+        "./snippets/tapewasm-86b4af57c39247cc/js/aot_bridge.js": import1,
+        "./snippets/tapewasm-86b4af57c39247cc/js/aot_bridge.js": import2,
+        "./snippets/tapewasm-86b4af57c39247cc/js/aot_bridge.js": import3,
+        "./snippets/tapewasm-86b4af57c39247cc/js/aot_bridge.js": import4,
+    };
+}
+
+const AotSamplerFinalization = (typeof FinalizationRegistry === 'undefined')
+    ? { register: () => {}, unregister: () => {} }
+    : new FinalizationRegistry(ptr => wasm.__wbg_aotsampler_free(ptr, 1));
+const CompiledTapeFinalization = (typeof FinalizationRegistry === 'undefined')
+    ? { register: () => {}, unregister: () => {} }
+    : new FinalizationRegistry(ptr => wasm.__wbg_compiledtape_free(ptr, 1));
+
+function addToExternrefTable0(obj) {
+    const idx = wasm.__externref_table_alloc();
+    wasm.__wbindgen_externrefs.set(idx, obj);
+    return idx;
+}
+
+function getArrayF64FromWasm0(ptr, len) {
+    ptr = ptr >>> 0;
+    return getFloat64ArrayMemory0().subarray(ptr / 8, ptr / 8 + len);
+}
+
+function getArrayU8FromWasm0(ptr, len) {
+    ptr = ptr >>> 0;
+    return getUint8ArrayMemory0().subarray(ptr / 1, ptr / 1 + len);
+}
+
+let cachedDataViewMemory0 = null;
+function getDataViewMemory0() {
+    if (cachedDataViewMemory0 === null || cachedDataViewMemory0.buffer.detached === true || (cachedDataViewMemory0.buffer.detached === undefined && cachedDataViewMemory0.buffer !== wasm.memory.buffer)) {
+        cachedDataViewMemory0 = new DataView(wasm.memory.buffer);
+    }
+    return cachedDataViewMemory0;
+}
+
+let cachedFloat64ArrayMemory0 = null;
+function getFloat64ArrayMemory0() {
+    if (cachedFloat64ArrayMemory0 === null || cachedFloat64ArrayMemory0.byteLength === 0) {
+        cachedFloat64ArrayMemory0 = new Float64Array(wasm.memory.buffer);
+    }
+    return cachedFloat64ArrayMemory0;
+}
+
+function getStringFromWasm0(ptr, len) {
+    return decodeText(ptr >>> 0, len);
+}
+
+let cachedUint8ArrayMemory0 = null;
+function getUint8ArrayMemory0() {
+    if (cachedUint8ArrayMemory0 === null || cachedUint8ArrayMemory0.byteLength === 0) {
+        cachedUint8ArrayMemory0 = new Uint8Array(wasm.memory.buffer);
+    }
+    return cachedUint8ArrayMemory0;
+}
+
+function isLikeNone(x) {
+    return x === undefined || x === null;
+}
+
+function passArrayF64ToWasm0(arg, malloc) {
+    const ptr = malloc(arg.length * 8, 8) >>> 0;
+    getFloat64ArrayMemory0().set(arg, ptr / 8);
+    WASM_VECTOR_LEN = arg.length;
+    return ptr;
+}
+
+function passArrayJsValueToWasm0(array, malloc) {
+    const ptr = malloc(array.length * 4, 4) >>> 0;
+    for (let i = 0; i < array.length; i++) {
+        const add = addToExternrefTable0(array[i]);
+        getDataViewMemory0().setUint32(ptr + 4 * i, add, true);
+    }
+    WASM_VECTOR_LEN = array.length;
+    return ptr;
+}
+
+function passStringToWasm0(arg, malloc, realloc) {
+    if (realloc === undefined) {
+        const buf = cachedTextEncoder.encode(arg);
+        const ptr = malloc(buf.length, 1) >>> 0;
+        getUint8ArrayMemory0().subarray(ptr, ptr + buf.length).set(buf);
+        WASM_VECTOR_LEN = buf.length;
+        return ptr;
+    }
+
+    let len = arg.length;
+    let ptr = malloc(len, 1) >>> 0;
+
+    const mem = getUint8ArrayMemory0();
+
+    let offset = 0;
+
+    for (; offset < len; offset++) {
+        const code = arg.charCodeAt(offset);
+        if (code > 0x7F) break;
+        mem[ptr + offset] = code;
+    }
+    if (offset !== len) {
+        if (offset !== 0) {
+            arg = arg.slice(offset);
+        }
+        ptr = realloc(ptr, len, len = offset + arg.length * 3, 1) >>> 0;
+        const view = getUint8ArrayMemory0().subarray(ptr + offset, ptr + len);
+        const ret = cachedTextEncoder.encodeInto(arg, view);
+
+        offset += ret.written;
+        ptr = realloc(ptr, len, offset, 1) >>> 0;
+    }
+
+    WASM_VECTOR_LEN = offset;
+    return ptr;
+}
+
+function takeFromExternrefTable0(idx) {
+    const value = wasm.__wbindgen_externrefs.get(idx);
+    wasm.__externref_table_dealloc(idx);
+    return value;
+}
+
+let cachedTextDecoder = new TextDecoder('utf-8', { ignoreBOM: true, fatal: true });
+cachedTextDecoder.decode();
+const MAX_SAFARI_DECODE_BYTES = 2146435072;
+let numBytesDecoded = 0;
+function decodeText(ptr, len) {
+    numBytesDecoded += len;
+    if (numBytesDecoded >= MAX_SAFARI_DECODE_BYTES) {
+        cachedTextDecoder = new TextDecoder('utf-8', { ignoreBOM: true, fatal: true });
+        cachedTextDecoder.decode();
+        numBytesDecoded = len;
+    }
+    return cachedTextDecoder.decode(getUint8ArrayMemory0().subarray(ptr, ptr + len));
+}
+
+const cachedTextEncoder = new TextEncoder();
+
+if (!('encodeInto' in cachedTextEncoder)) {
+    cachedTextEncoder.encodeInto = function (arg, view) {
+        const buf = cachedTextEncoder.encode(arg);
+        view.set(buf);
+        return {
+            read: arg.length,
+            written: buf.length
+        };
+    };
+}
+
+let WASM_VECTOR_LEN = 0;
+
+let wasmModule, wasmInstance, wasm;
+function __wbg_finalize_init(instance, module) {
+    wasmInstance = instance;
+    wasm = instance.exports;
+    wasmModule = module;
+    cachedDataViewMemory0 = null;
+    cachedFloat64ArrayMemory0 = null;
+    cachedUint8ArrayMemory0 = null;
+    wasm.__wbindgen_start();
+    return wasm;
+}
+
+async function __wbg_load(module, imports) {
+    if (typeof Response === 'function' && module instanceof Response) {
+        if (!module.ok) {
+            throw new Error(`failed to fetch Wasm: ${module.status} ${module.statusText} fetching '${module.url}'`);
+        }
+
+        if (typeof WebAssembly.instantiateStreaming === 'function') {
+            try {
+                return await WebAssembly.instantiateStreaming(module, imports);
+            } catch (e) {
+                const validResponse = expectedResponseType(module.type);
+
+                if (validResponse && module.headers.get('Content-Type') !== 'application/wasm') {
+                    console.warn("`WebAssembly.instantiateStreaming` failed because your server does not serve Wasm with `application/wasm` MIME type. Falling back to `WebAssembly.instantiate` which is slower. Original error:\n", e);
+
+                } else { throw e; }
+            }
+        }
+
+        const bytes = await module.arrayBuffer();
+        return await WebAssembly.instantiate(bytes, imports);
+    } else {
+        const instance = await WebAssembly.instantiate(module, imports);
+
+        if (instance instanceof WebAssembly.Instance) {
+            return { instance, module };
+        } else {
+            return instance;
+        }
+    }
+
+    function expectedResponseType(type) {
+        switch (type) {
+            case 'basic': case 'cors': case 'default': return true;
+        }
+        return false;
+    }
+}
+
+function initSync(module) {
+    if (wasm !== undefined) return wasm;
+
+
+    if (module !== undefined) {
+        if (Object.getPrototypeOf(module) === Object.prototype) {
+            ({module} = module)
+        } else {
+            console.warn('using deprecated parameters for `initSync()`; pass a single object instead')
+        }
+    }
+
+    const imports = __wbg_get_imports();
+    if (!(module instanceof WebAssembly.Module)) {
+        module = new WebAssembly.Module(module);
+    }
+    const instance = new WebAssembly.Instance(module, imports);
+    return __wbg_finalize_init(instance, module);
+}
+
+async function __wbg_init(module_or_path) {
+    if (wasm !== undefined) return wasm;
+
+
+    if (module_or_path !== undefined) {
+        if (Object.getPrototypeOf(module_or_path) === Object.prototype) {
+            ({module_or_path} = module_or_path)
+        } else {
+            console.warn('using deprecated parameters for the initialization function; pass a single object instead')
+        }
+    }
+
+    if (module_or_path === undefined) {
+        module_or_path = new URL('tapewasm_bg.wasm', import.meta.url);
+    }
+    const imports = __wbg_get_imports();
+
+    if (typeof module_or_path === 'string' || (typeof Request === 'function' && module_or_path instanceof Request) || (typeof URL === 'function' && module_or_path instanceof URL)) {
+        module_or_path = fetch(module_or_path);
+    }
+
+    const { instance, module } = await __wbg_load(await module_or_path, imports);
+
+    return __wbg_finalize_init(instance, module);
+}
+
+export { initSync, __wbg_init as default };

package/pkg/tapewasm_bg.wasm.d.ts

@@ -0,0 +1,26 @@
+/* tslint:disable */
+/* eslint-disable */
+export const memory: WebAssembly.Memory;
+export const __wbg_aotsampler_free: (a: number, b: number) => void;
+export const __wbg_compiledtape_free: (a: number, b: number) => void;
+export const aotsampler_logProbGrad: (a: number, b: number, c: number) => [number, number, number, number];
+export const aotsampler_nParams: (a: number) => number;
+export const aotsampler_new: (a: number, b: number, c: number, d: number, e: number, f: number) => [number, number, number];
+export const aotsampler_sample: (a: number, b: number, c: number, d: number, e: number, f: bigint) => [number, number, number, number];
+export const clearAotExports: () => void;
+export const compileTape: (a: number, b: number) => [number, number, number];
+export const compiledtape_layoutId: (a: number) => number;
+export const compiledtape_nParams: (a: number) => number;
+export const compiledtape_scratchInit: (a: number) => [number, number];
+export const compiledtape_wasm: (a: number) => [number, number];
+export const init_panic_hook: () => void;
+export const setAotExports: (a: any) => void;
+export const sharedMemory: () => any;
+export const tapewasmVersion: () => [number, number];
+export const __wbindgen_malloc: (a: number, b: number) => number;
+export const __wbindgen_realloc: (a: number, b: number, c: number, d: number) => number;
+export const __wbindgen_free: (a: number, b: number, c: number) => void;
+export const __wbindgen_externrefs: WebAssembly.Table;
+export const __externref_table_dealloc: (a: number) => void;
+export const __externref_table_alloc: () => number;
+export const __wbindgen_start: () => void;