tapewasm 0.3.4 → 0.3.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "tapewasm",
-  "version": "0.3.4",
+  "version": "0.3.5",
   "description": "Compile an autodiff tape to a WebAssembly module and sample it with nuts-rs, in the browser.",
   "type": "module",
   "main": "./index.js",

package/pkg/package.json

@@ -2,7 +2,7 @@
   "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.3.4",
+  "version": "0.3.5",
   "license": "MIT OR Apache-2.0",
   "repository": {
     "type": "git",

package/pkg/tapewasm.d.ts

@@ -116,6 +116,13 @@ export class AotSampler {
      * posteriors; see [`nuts_settings`].
      */
     setGradBasedEstimate(on: boolean): void;
+    /**
+     * Keep each trajectory below `2^depth` leapfrog steps instead of nuts-rs's
+     * 2^10. Lower bounds the gradients per draw at the risk of stopping before
+     * the trajectory turns, which the draws' `numSteps` then show as a flat top.
+     * A number rather than a `u32`, which JS would wrap (-1) or truncate (2.7).
+     */
+    setMaxDepth(depth: number): void;
     /**
      * Aim warmup's step-size adaptation at this acceptance rate instead of
      * nuts-rs's 0.8. Higher adapts a smaller step: fewer divergences on a hard
@@ -259,6 +266,7 @@ export interface InitOutput {
     readonly aotsampler_sample: (a: number, b: number, c: number, d: number, e: number, f: bigint) => [number, number, number, number];
     readonly aotsampler_sampleWithStats: (a: number, b: number, c: number, d: number, e: number, f: bigint, g: number) => [number, number, number];
     readonly aotsampler_setGradBasedEstimate: (a: number, b: number) => void;
+    readonly aotsampler_setMaxDepth: (a: number, b: number) => [number, number];
     readonly aotsampler_setTargetAccept: (a: number, b: number) => [number, number];
     readonly clearAotExports: () => void;
     readonly compileTape: (a: number, b: number, c: number, d: number) => [number, number, number];

package/pkg/tapewasm.js

@@ -1,10 +1,10 @@
 /* @ts-self-types="./tapewasm.d.ts" */
-import { aot_evaluate, aot_logp } from './snippets/tapewasm-a36239fb19e3c5d4/js/aot_bridge.js';
-import * as import1 from "./snippets/tapewasm-a36239fb19e3c5d4/js/aot_bridge.js"
-import * as import2 from "./snippets/tapewasm-a36239fb19e3c5d4/js/aot_bridge.js"
-import * as import3 from "./snippets/tapewasm-a36239fb19e3c5d4/js/aot_bridge.js"
-import * as import4 from "./snippets/tapewasm-a36239fb19e3c5d4/js/aot_bridge.js"
-import * as import5 from "./snippets/tapewasm-a36239fb19e3c5d4/js/aot_bridge.js"
+import { aot_evaluate, aot_logp } from './snippets/tapewasm-8f332cef44599c8c/js/aot_bridge.js';
+import * as import1 from "./snippets/tapewasm-8f332cef44599c8c/js/aot_bridge.js"
+import * as import2 from "./snippets/tapewasm-8f332cef44599c8c/js/aot_bridge.js"
+import * as import3 from "./snippets/tapewasm-8f332cef44599c8c/js/aot_bridge.js"
+import * as import4 from "./snippets/tapewasm-8f332cef44599c8c/js/aot_bridge.js"
+import * as import5 from "./snippets/tapewasm-8f332cef44599c8c/js/aot_bridge.js"
 
 
 /**
@@ -273,6 +273,19 @@ export class AotSampler {
     setGradBasedEstimate(on) {
         wasm.aotsampler_setGradBasedEstimate(this.__wbg_ptr, on);
     }
+    /**
+     * Keep each trajectory below `2^depth` leapfrog steps instead of nuts-rs's
+     * 2^10. Lower bounds the gradients per draw at the risk of stopping before
+     * the trajectory turns, which the draws' `numSteps` then show as a flat top.
+     * A number rather than a `u32`, which JS would wrap (-1) or truncate (2.7).
+     * @param {number} depth
+     */
+    setMaxDepth(depth) {
+        const ret = wasm.aotsampler_setMaxDepth(this.__wbg_ptr, depth);
+        if (ret[1]) {
+            throw takeFromExternrefTable0(ret[0]);
+        }
+    }
     /**
      * Aim warmup's step-size adaptation at this acceptance rate instead of
      * nuts-rs's 0.8. Higher adapts a smaller step: fewer divergences on a hard
@@ -563,15 +576,15 @@ function __wbg_get_imports() {
         __wbg___wbindgen_throw_5d9e815e6fdf150f: function(arg0, arg1) {
             throw new Error(getStringFromWasm0(arg0, arg1));
         },
-        __wbg_aot_evaluate_85ed206dc7e5ceef: function() { return handleError(function (arg0, arg1, arg2, arg3) {
+        __wbg_aot_evaluate_59e3797ad39ed69a: function() { return handleError(function (arg0, arg1, arg2, arg3) {
             const ret = aot_evaluate(arg0 >>> 0, arg1 >>> 0, arg2 >>> 0, arg3 >>> 0);
             return ret;
         }, arguments); },
-        __wbg_aot_logp_845ba7cb025408ad: function(arg0, arg1, arg2, arg3) {
+        __wbg_aot_logp_05b8eff6985b4538: function(arg0, arg1, arg2, arg3) {
             const ret = aot_logp(arg0 >>> 0, arg1 >>> 0, arg2 >>> 0, arg3 >>> 0);
             return ret;
         },
-        __wbg_call_81c6c30636ec594e: function() { return handleError(function (arg0, arg1, arg2, arg3, arg4, arg5, arg6) {
+        __wbg_call_e22c9e5bbebdaf41: function() { return handleError(function (arg0, arg1, arg2, arg3, arg4, arg5, arg6) {
             var v0 = getArrayF64FromWasm0(arg3, arg4).slice();
             wasm.__wbindgen_free(arg3, arg4 * 8, 8);
             var v1 = getArrayF64FromWasm0(arg5, arg6).slice();
@@ -614,11 +627,11 @@ function __wbg_get_imports() {
     return {
         __proto__: null,
         "./tapewasm_bg.js": import0,
-        "./snippets/tapewasm-a36239fb19e3c5d4/js/aot_bridge.js": import1,
-        "./snippets/tapewasm-a36239fb19e3c5d4/js/aot_bridge.js": import2,
-        "./snippets/tapewasm-a36239fb19e3c5d4/js/aot_bridge.js": import3,
-        "./snippets/tapewasm-a36239fb19e3c5d4/js/aot_bridge.js": import4,
-        "./snippets/tapewasm-a36239fb19e3c5d4/js/aot_bridge.js": import5,
+        "./snippets/tapewasm-8f332cef44599c8c/js/aot_bridge.js": import1,
+        "./snippets/tapewasm-8f332cef44599c8c/js/aot_bridge.js": import2,
+        "./snippets/tapewasm-8f332cef44599c8c/js/aot_bridge.js": import3,
+        "./snippets/tapewasm-8f332cef44599c8c/js/aot_bridge.js": import4,
+        "./snippets/tapewasm-8f332cef44599c8c/js/aot_bridge.js": import5,
     };
 }
 

package/pkg/tapewasm_bg.wasm.d.ts

@@ -18,6 +18,7 @@ export const aotsampler_new: (a: number, b: number, c: number, d: number, e: num
 export const aotsampler_sample: (a: number, b: number, c: number, d: number, e: number, f: bigint) => [number, number, number, number];
 export const aotsampler_sampleWithStats: (a: number, b: number, c: number, d: number, e: number, f: bigint, g: number) => [number, number, number];
 export const aotsampler_setGradBasedEstimate: (a: number, b: number) => void;
+export const aotsampler_setMaxDepth: (a: number, b: number) => [number, number];
 export const aotsampler_setTargetAccept: (a: number, b: number) => [number, number];
 export const clearAotExports: () => void;
 export const compileTape: (a: number, b: number, c: number, d: number) => [number, number, number];