statedict2pytree 0.5.2__py3-none-any.whl → 0.5.4__py3-none-any.whl

client/public/bundle.js

@@ -116,6 +116,12 @@ var app = (function () {
 		return text(' ');
 	}
 
+	/**
+	 * @returns {Text} */
+	function empty() {
+		return text('');
+	}
+
 	/**
 	 * @param {EventTarget} node
 	 * @param {string} event
@@ -153,6 +159,26 @@ var app = (function () {
 		input.value = value == null ? '' : value;
 	}
 
+	/**
+	 * @returns {void} */
+	function select_option(select, value, mounting) {
+		for (let i = 0; i < select.options.length; i += 1) {
+			const option = select.options[i];
+			if (option.__value === value) {
+				option.selected = true;
+				return;
+			}
+		}
+		if (!mounting || value !== undefined) {
+			select.selectedIndex = -1; // no option should be selected
+		}
+	}
+
+	function select_value(select) {
+		const selected_option = select.querySelector(':checked');
+		return selected_option && selected_option.__value;
+	}
+
 	/**
 	 * @template T
 	 * @param {string} type
@@ -8737,26 +8763,26 @@ var app = (function () {
 
 	function get_each_context(ctx, list, i) {
 		const child_ctx = ctx.slice();
-		child_ctx[14] = list[i];
-		child_ctx[16] = i;
+		child_ctx[18] = list[i];
+		child_ctx[20] = i;
 		return child_ctx;
 	}
 
 	function get_each_context_1(ctx, list, i) {
 		const child_ctx = ctx.slice();
-		child_ctx[14] = list[i];
-		child_ctx[16] = i;
+		child_ctx[18] = list[i];
+		child_ctx[20] = i;
 		return child_ctx;
 	}
 
-	// (231:12) {#each jaxFields as field, i}
+	// (458:12) {#each jaxFields as field, i}
 	function create_each_block_1(ctx) {
 		let div1;
 		let div0;
-		let t0_value = /*field*/ ctx[14].path + "";
+		let t0_value = /*field*/ ctx[18].path + "";
 		let t0;
 		let t1;
-		let t2_value = /*field*/ ctx[14].shape + "";
+		let t2_value = /*field*/ ctx[18].shape + "";
 		let t2;
 		let div0_data_path_value;
 		let div0_data_shape_value;
@@ -8772,16 +8798,16 @@ var app = (function () {
 				t1 = space();
 				t2 = text(t2_value);
 				t3 = space();
-				attr_dev(div0, "id", "jax-" + String(/*i*/ ctx[16]));
+				attr_dev(div0, "id", "jax-" + String(/*i*/ ctx[20]));
 				attr_dev(div0, "class", "whitespace-nowrap overflow-x-scroll cursor-pointer mx-2");
 				attr_dev(div0, "data-jax", "jax");
-				attr_dev(div0, "data-path", div0_data_path_value = /*field*/ ctx[14].path);
-				attr_dev(div0, "data-shape", div0_data_shape_value = /*field*/ ctx[14].shape);
-				attr_dev(div0, "data-skip", div0_data_skip_value = /*field*/ ctx[14].skip);
-				attr_dev(div0, "data-type", div0_data_type_value = /*field*/ ctx[14].type);
-				add_location(div0, file, 268, 20, 7693);
+				attr_dev(div0, "data-path", div0_data_path_value = /*field*/ ctx[18].path);
+				attr_dev(div0, "data-shape", div0_data_shape_value = /*field*/ ctx[18].shape);
+				attr_dev(div0, "data-skip", div0_data_skip_value = /*field*/ ctx[18].skip);
+				attr_dev(div0, "data-type", div0_data_type_value = /*field*/ ctx[18].type);
+				add_location(div0, file, 506, 20, 13315);
 				attr_dev(div1, "class", "border h-12 rounded-xl flex flex-col justify-center");
-				add_location(div1, file, 265, 16, 7570);
+				add_location(div1, file, 503, 16, 13192);
 			},
 			m: function mount(target, anchor) {
 				insert_dev(target, div1, anchor);
@@ -8792,22 +8818,22 @@ var app = (function () {
 				append_dev(div1, t3);
 			},
 			p: function update(ctx, dirty) {
-				if (dirty & /*jaxFields*/ 2 && t0_value !== (t0_value = /*field*/ ctx[14].path + "")) set_data_dev(t0, t0_value);
-				if (dirty & /*jaxFields*/ 2 && t2_value !== (t2_value = /*field*/ ctx[14].shape + "")) set_data_dev(t2, t2_value);
+				if (dirty & /*jaxFields*/ 4 && t0_value !== (t0_value = /*field*/ ctx[18].path + "")) set_data_dev(t0, t0_value);
+				if (dirty & /*jaxFields*/ 4 && t2_value !== (t2_value = /*field*/ ctx[18].shape + "")) set_data_dev(t2, t2_value);
 
-				if (dirty & /*jaxFields*/ 2 && div0_data_path_value !== (div0_data_path_value = /*field*/ ctx[14].path)) {
+				if (dirty & /*jaxFields*/ 4 && div0_data_path_value !== (div0_data_path_value = /*field*/ ctx[18].path)) {
 					attr_dev(div0, "data-path", div0_data_path_value);
 				}
 
-				if (dirty & /*jaxFields*/ 2 && div0_data_shape_value !== (div0_data_shape_value = /*field*/ ctx[14].shape)) {
+				if (dirty & /*jaxFields*/ 4 && div0_data_shape_value !== (div0_data_shape_value = /*field*/ ctx[18].shape)) {
 					attr_dev(div0, "data-shape", div0_data_shape_value);
 				}
 
-				if (dirty & /*jaxFields*/ 2 && div0_data_skip_value !== (div0_data_skip_value = /*field*/ ctx[14].skip)) {
+				if (dirty & /*jaxFields*/ 4 && div0_data_skip_value !== (div0_data_skip_value = /*field*/ ctx[18].skip)) {
 					attr_dev(div0, "data-skip", div0_data_skip_value);
 				}
 
-				if (dirty & /*jaxFields*/ 2 && div0_data_type_value !== (div0_data_type_value = /*field*/ ctx[14].type)) {
+				if (dirty & /*jaxFields*/ 4 && div0_data_type_value !== (div0_data_type_value = /*field*/ ctx[18].type)) {
 					attr_dev(div0, "data-type", div0_data_type_value);
 				}
 			},
@@ -8822,19 +8848,19 @@ var app = (function () {
 			block,
 			id: create_each_block_1.name,
 			type: "each",
-			source: "(231:12) {#each jaxFields as field, i}",
+			source: "(458:12) {#each jaxFields as field, i}",
 			ctx
 		});
 
 		return block;
 	}
 
-	// (268:24) {:else}
+	// (496:28) {:else}
 	function create_else_block(ctx) {
-		let t0_value = /*field*/ ctx[14].path + "";
+		let t0_value = /*field*/ ctx[18].path + "";
 		let t0;
 		let t1;
-		let t2_value = /*field*/ ctx[14].shape + "";
+		let t2_value = /*field*/ ctx[18].shape + "";
 		let t2;
 
 		const block = {
@@ -8849,8 +8875,8 @@ var app = (function () {
 				insert_dev(target, t2, anchor);
 			},
 			p: function update(ctx, dirty) {
-				if (dirty & /*torchFields*/ 4 && t0_value !== (t0_value = /*field*/ ctx[14].path + "")) set_data_dev(t0, t0_value);
-				if (dirty & /*torchFields*/ 4 && t2_value !== (t2_value = /*field*/ ctx[14].shape + "")) set_data_dev(t2, t2_value);
+				if (dirty & /*torchFields*/ 8 && t0_value !== (t0_value = /*field*/ ctx[18].path + "")) set_data_dev(t0, t0_value);
+				if (dirty & /*torchFields*/ 8 && t2_value !== (t2_value = /*field*/ ctx[18].shape + "")) set_data_dev(t2, t2_value);
 			},
 			d: function destroy(detaching) {
 				if (detaching) {
@@ -8865,14 +8891,14 @@ var app = (function () {
 			block,
 			id: create_else_block.name,
 			type: "else",
-			source: "(268:24) {:else}",
+			source: "(496:28) {:else}",
 			ctx
 		});
 
 		return block;
 	}
 
-	// (266:24) {#if field.skip}
+	// (494:28) {#if field.skip}
 	function create_if_block_1(ctx) {
 		let t;
 
@@ -8895,21 +8921,21 @@ var app = (function () {
 			block,
 			id: create_if_block_1.name,
 			type: "if",
-			source: "(266:24) {#if field.skip}",
+			source: "(494:28) {#if field.skip}",
 			ctx
 		});
 
 		return block;
 	}
 
-	// (273:20) {#if field.skip}
+	// (501:24) {#if field.skip}
 	function create_if_block(ctx) {
 		let button;
 		let mounted;
 		let dispose;
 
 		function click_handler() {
-			return /*click_handler*/ ctx[6](/*i*/ ctx[16]);
+			return /*click_handler*/ ctx[11](/*i*/ ctx[20]);
 		}
 
 		const block = {
@@ -8917,7 +8943,7 @@ var app = (function () {
 				button = element("button");
 				button.textContent = "-";
 				attr_dev(button, "class", "btn btn-ghost");
-				add_location(button, file, 307, 24, 9211);
+				add_location(button, file, 546, 28, 14944);
 			},
 			m: function mount(target, anchor) {
 				insert_dev(target, button, anchor);
@@ -8944,14 +8970,14 @@ var app = (function () {
 			block,
 			id: create_if_block.name,
 			type: "if",
-			source: "(273:20) {#if field.skip}",
+			source: "(501:24) {#if field.skip}",
 			ctx
 		});
 
 		return block;
 	}
 
-	// (255:12) {#each torchFields as field, i}
+	// (483:16) {#each torchFields as field, i}
 	function create_each_block(ctx) {
 		let div1;
 		let div0;
@@ -8967,16 +8993,16 @@ var app = (function () {
 		let dispose;
 
 		function select_block_type(ctx, dirty) {
-			if (/*field*/ ctx[14].skip) return create_if_block_1;
+			if (/*field*/ ctx[18].skip) return create_if_block_1;
 			return create_else_block;
 		}
 
 		let current_block_type = select_block_type(ctx);
 		let if_block0 = current_block_type(ctx);
-		let if_block1 = /*field*/ ctx[14].skip && create_if_block(ctx);
+		let if_block1 = /*field*/ ctx[18].skip && create_if_block(ctx);
 
 		function click_handler_1() {
-			return /*click_handler_1*/ ctx[7](/*i*/ ctx[16]);
+			return /*click_handler_1*/ ctx[12](/*i*/ ctx[20]);
 		}
 
 		const block = {
@@ -8990,18 +9016,18 @@ var app = (function () {
 				button = element("button");
 				button.textContent = "+";
 				t3 = space();
-				attr_dev(div0, "id", "torch-" + String(/*i*/ ctx[16]));
+				attr_dev(div0, "id", "torch-" + String(/*i*/ ctx[20]));
 				attr_dev(div0, "data-torch", "torch");
-				attr_dev(div0, "data-path", div0_data_path_value = /*field*/ ctx[14].path);
-				attr_dev(div0, "data-shape", div0_data_shape_value = /*field*/ ctx[14].shape);
-				attr_dev(div0, "data-skip", div0_data_skip_value = /*field*/ ctx[14].skip);
-				attr_dev(div0, "data-type", div0_data_type_value = /*field*/ ctx[14].type);
+				attr_dev(div0, "data-path", div0_data_path_value = /*field*/ ctx[18].path);
+				attr_dev(div0, "data-shape", div0_data_shape_value = /*field*/ ctx[18].shape);
+				attr_dev(div0, "data-skip", div0_data_skip_value = /*field*/ ctx[18].skip);
+				attr_dev(div0, "data-type", div0_data_type_value = /*field*/ ctx[18].type);
 				attr_dev(div0, "class", "flex-1 mx-2 my-auto whitespace-nowrap overflow-x-scroll cursor-pointer");
-				add_location(div0, file, 290, 20, 8491);
+				add_location(div0, file, 529, 24, 14156);
 				attr_dev(button, "class", "btn btn-ghost");
-				add_location(button, file, 314, 20, 9480);
+				add_location(button, file, 553, 24, 15241);
 				attr_dev(div1, "class", "flex space-x-2 border h-12 rounded-xl");
-				add_location(div1, file, 289, 16, 8419);
+				add_location(div1, file, 528, 20, 14080);
 			},
 			m: function mount(target, anchor) {
 				insert_dev(target, div1, anchor);
@@ -9033,23 +9059,23 @@ var app = (function () {
 					}
 				}
 
-				if (dirty & /*torchFields*/ 4 && div0_data_path_value !== (div0_data_path_value = /*field*/ ctx[14].path)) {
+				if (dirty & /*torchFields*/ 8 && div0_data_path_value !== (div0_data_path_value = /*field*/ ctx[18].path)) {
 					attr_dev(div0, "data-path", div0_data_path_value);
 				}
 
-				if (dirty & /*torchFields*/ 4 && div0_data_shape_value !== (div0_data_shape_value = /*field*/ ctx[14].shape)) {
+				if (dirty & /*torchFields*/ 8 && div0_data_shape_value !== (div0_data_shape_value = /*field*/ ctx[18].shape)) {
 					attr_dev(div0, "data-shape", div0_data_shape_value);
 				}
 
-				if (dirty & /*torchFields*/ 4 && div0_data_skip_value !== (div0_data_skip_value = /*field*/ ctx[14].skip)) {
+				if (dirty & /*torchFields*/ 8 && div0_data_skip_value !== (div0_data_skip_value = /*field*/ ctx[18].skip)) {
 					attr_dev(div0, "data-skip", div0_data_skip_value);
 				}
 
-				if (dirty & /*torchFields*/ 4 && div0_data_type_value !== (div0_data_type_value = /*field*/ ctx[14].type)) {
+				if (dirty & /*torchFields*/ 8 && div0_data_type_value !== (div0_data_type_value = /*field*/ ctx[18].type)) {
 					attr_dev(div0, "data-type", div0_data_type_value);
 				}
 
-				if (/*field*/ ctx[14].skip) {
+				if (/*field*/ ctx[18].skip) {
 					if (if_block1) {
 						if_block1.p(ctx, dirty);
 					} else {
@@ -9078,7 +9104,78 @@ var app = (function () {
 			block,
 			id: create_each_block.name,
 			type: "each",
-			source: "(255:12) {#each torchFields as field, i}",
+			source: "(483:16) {#each torchFields as field, i}",
+			ctx
+		});
+
+		return block;
+	}
+
+	// (482:12) {#key torchFields}
+	function create_key_block(ctx) {
+		let each_1_anchor;
+		let each_value = ensure_array_like_dev(/*torchFields*/ ctx[3]);
+		let each_blocks = [];
+
+		for (let i = 0; i < each_value.length; i += 1) {
+			each_blocks[i] = create_each_block(get_each_context(ctx, each_value, i));
+		}
+
+		const block = {
+			c: function create() {
+				for (let i = 0; i < each_blocks.length; i += 1) {
+					each_blocks[i].c();
+				}
+
+				each_1_anchor = empty();
+			},
+			m: function mount(target, anchor) {
+				for (let i = 0; i < each_blocks.length; i += 1) {
+					if (each_blocks[i]) {
+						each_blocks[i].m(target, anchor);
+					}
+				}
+
+				insert_dev(target, each_1_anchor, anchor);
+			},
+			p: function update(ctx, dirty) {
+				if (dirty & /*addSkipLayer, removeSkipLayer, torchFields, String*/ 56) {
+					each_value = ensure_array_like_dev(/*torchFields*/ ctx[3]);
+					let i;
+
+					for (i = 0; i < each_value.length; i += 1) {
+						const child_ctx = get_each_context(ctx, each_value, i);
+
+						if (each_blocks[i]) {
+							each_blocks[i].p(child_ctx, dirty);
+						} else {
+							each_blocks[i] = create_each_block(child_ctx);
+							each_blocks[i].c();
+							each_blocks[i].m(each_1_anchor.parentNode, each_1_anchor);
+						}
+					}
+
+					for (; i < each_blocks.length; i += 1) {
+						each_blocks[i].d(1);
+					}
+
+					each_blocks.length = each_value.length;
+				}
+			},
+			d: function destroy(detaching) {
+				if (detaching) {
+					detach_dev(each_1_anchor);
+				}
+
+				destroy_each(each_blocks, detaching);
+			}
+		};
+
+		dispatch_dev("SvelteRegisterBlock", {
+			block,
+			id: create_key_block.name,
+			type: "key",
+			source: "(482:12) {#key torchFields}",
 			ctx
 		});
 
@@ -9089,37 +9186,46 @@ var app = (function () {
 		let t0;
 		let h1;
 		let t2;
-		let div4;
 		let div1;
-		let h20;
+		let button0;
 		let t4;
+		let button1;
+		let t6;
 		let div0;
-		let t5;
-		let div3;
-		let h21;
-		let t7;
-		let div2;
+		let button2;
 		let t8;
+		let select;
+		let option0;
+		let option1;
+		let option2;
+		let t12;
 		let div6;
+		let div3;
+		let h20;
+		let t14;
+		let div2;
+		let t15;
 		let div5;
+		let h21;
+		let t17;
+		let div4;
+		let previous_key = /*torchFields*/ ctx[3];
+		let t18;
+		let div8;
+		let div7;
 		let input;
-		let t9;
-		let button;
+		let t19;
+		let button3;
 		let mounted;
 		let dispose;
-		let each_value_1 = ensure_array_like_dev(/*jaxFields*/ ctx[1]);
-		let each_blocks_1 = [];
+		let each_value_1 = ensure_array_like_dev(/*jaxFields*/ ctx[2]);
+		let each_blocks = [];
 
 		for (let i = 0; i < each_value_1.length; i += 1) {
-			each_blocks_1[i] = create_each_block_1(get_each_context_1(ctx, each_value_1, i));
+			each_blocks[i] = create_each_block_1(get_each_context_1(ctx, each_value_1, i));
 		}
 
-		let each_value = ensure_array_like_dev(/*torchFields*/ ctx[2]);
-		let each_blocks = [];
-
-		for (let i = 0; i < each_value.length; i += 1) {
-			each_blocks[i] = create_each_block(get_each_context(ctx, each_value, i));
-		}
+		let key_block = create_key_block(ctx);
 
 		const block = {
 			c: function create() {
@@ -9127,66 +9233,101 @@ var app = (function () {
 				h1 = element("h1");
 				h1.textContent = "Welcome to Torch2Jax";
 				t2 = space();
-				div4 = element("div");
 				div1 = element("div");
-				h20 = element("h2");
-				h20.textContent = "JAX";
+				button0 = element("button");
+				button0.textContent = "Pad To Match";
 				t4 = space();
+				button1 = element("button");
+				button1.textContent = "Remove All Skip Layers";
+				t6 = space();
 				div0 = element("div");
-
-				for (let i = 0; i < each_blocks_1.length; i += 1) {
-					each_blocks_1[i].c();
-				}
-
-				t5 = space();
+				button2 = element("button");
+				button2.textContent = "Match By Name";
+				t8 = space();
+				select = element("select");
+				option0 = element("option");
+				option0.textContent = "opus";
+				option1 = element("option");
+				option1.textContent = "sonnet";
+				option2 = element("option");
+				option2.textContent = "haiku";
+				t12 = space();
+				div6 = element("div");
 				div3 = element("div");
-				h21 = element("h2");
-				h21.textContent = "PyTorch";
-				t7 = space();
+				h20 = element("h2");
+				h20.textContent = "JAX";
+				t14 = space();
 				div2 = element("div");
 
 				for (let i = 0; i < each_blocks.length; i += 1) {
 					each_blocks[i].c();
 				}
 
-				t8 = space();
-				div6 = element("div");
+				t15 = space();
 				div5 = element("div");
+				h21 = element("h2");
+				h21.textContent = "PyTorch";
+				t17 = space();
+				div4 = element("div");
+				key_block.c();
+				t18 = space();
+				div8 = element("div");
+				div7 = element("div");
 				input = element("input");
-				t9 = space();
-				button = element("button");
-				button.textContent = "Convert!";
+				t19 = space();
+				button3 = element("button");
+				button3.textContent = "Convert!";
 				document_1.title = "Statedict2PyTree";
 				attr_dev(h1, "class", "text-3xl my-12");
-				add_location(h1, file, 258, 0, 7323);
+				add_location(h1, file, 481, 0, 12367);
+				attr_dev(button0, "class", "btn btn-accent");
+				add_location(button0, file, 483, 4, 12463);
+				attr_dev(button1, "class", "btn btn-secondary");
+				add_location(button1, file, 484, 4, 12542);
+				attr_dev(button2, "class", "btn btn-warning");
+				add_location(button2, file, 488, 8, 12671);
+				option0.__value = "opus";
+				set_input_value(option0, option0.__value);
+				add_location(option0, file, 492, 12, 12828);
+				option1.__value = "sonnet";
+				set_input_value(option1, option1.__value);
+				add_location(option1, file, 493, 12, 12875);
+				option2.__value = "haiku";
+				set_input_value(option2, option2.__value);
+				add_location(option2, file, 494, 12, 12926);
+				if (/*anthropicModel*/ ctx[1] === void 0) add_render_callback(() => /*select_change_handler*/ ctx[10].call(select));
+				add_location(select, file, 491, 8, 12779);
+				add_location(div0, file, 487, 4, 12657);
+				attr_dev(div1, "class", "my-4 flex justify-evenly");
+				add_location(div1, file, 482, 0, 12420);
 				attr_dev(h20, "class", "text-2xl");
-				add_location(h20, file, 262, 8, 7443);
-				attr_dev(div0, "id", "jax-fields");
-				attr_dev(div0, "class", "");
-				add_location(div0, file, 263, 8, 7481);
-				attr_dev(div1, "class", "");
-				add_location(div1, file, 261, 4, 7420);
-				attr_dev(h21, "class", "text-2xl");
-				add_location(h21, file, 286, 8, 8284);
-				attr_dev(div2, "id", "torch-fields");
+				add_location(h20, file, 500, 8, 13065);
+				attr_dev(div2, "id", "jax-fields");
 				attr_dev(div2, "class", "");
-				add_location(div2, file, 287, 8, 8326);
+				add_location(div2, file, 501, 8, 13103);
 				attr_dev(div3, "class", "");
-				add_location(div3, file, 285, 4, 8261);
-				attr_dev(div4, "class", "grid grid-cols-2 gap-x-2");
-				add_location(div4, file, 260, 0, 7377);
+				add_location(div3, file, 499, 4, 13042);
+				attr_dev(h21, "class", "text-2xl");
+				add_location(h21, file, 524, 8, 13906);
+				attr_dev(div4, "id", "torch-fields");
+				attr_dev(div4, "class", "");
+				add_location(div4, file, 525, 8, 13948);
+				attr_dev(div5, "class", "");
+				add_location(div5, file, 523, 4, 13883);
+				attr_dev(div6, "class", "grid grid-cols-2 gap-x-2");
+				add_location(div6, file, 498, 0, 12999);
 				attr_dev(input, "id", "name");
 				attr_dev(input, "type", "text");
 				attr_dev(input, "name", "name");
 				attr_dev(input, "class", "input input-primary w-full");
 				attr_dev(input, "placeholder", "Name of the new file (model.eqx per default)");
-				add_location(input, file, 327, 8, 9865);
-				attr_dev(button, "class", "btn btn-accent btn-wide btn-lg mx-auto my-2");
-				add_location(button, file, 335, 8, 10110);
-				attr_dev(div5, "class", "flex flex-col justify-center w-full");
-				add_location(div5, file, 326, 4, 9807);
-				attr_dev(div6, "class", "flex justify-center my-12 w-full");
-				add_location(div6, file, 325, 0, 9756);
+				add_location(input, file, 567, 8, 15673);
+				attr_dev(button3, "class", "btn btn-accent btn-wide btn-lg mx-auto my-2");
+				add_location(button3, file, 575, 8, 15918);
+				attr_dev(div7, "class", "flex flex-col justify-center w-full");
+				add_location(div7, file, 566, 4, 15615);
+				attr_dev(div8, "class", "flex justify-center my-12 w-full");
+				add_location(div8, file, 565, 0, 15564);
 			},
 			l: function claim(nodes) {
 				throw new Error("options.hydrate only works if the component was compiled with the `hydratable: true` option");
@@ -9195,22 +9336,24 @@ var app = (function () {
 				insert_dev(target, t0, anchor);
 				insert_dev(target, h1, anchor);
 				insert_dev(target, t2, anchor);
-				insert_dev(target, div4, anchor);
-				append_dev(div4, div1);
-				append_dev(div1, h20);
+				insert_dev(target, div1, anchor);
+				append_dev(div1, button0);
 				append_dev(div1, t4);
+				append_dev(div1, button1);
+				append_dev(div1, t6);
 				append_dev(div1, div0);
-
-				for (let i = 0; i < each_blocks_1.length; i += 1) {
-					if (each_blocks_1[i]) {
-						each_blocks_1[i].m(div0, null);
-					}
-				}
-
-				append_dev(div4, t5);
-				append_dev(div4, div3);
-				append_dev(div3, h21);
-				append_dev(div3, t7);
+				append_dev(div0, button2);
+				append_dev(div0, t8);
+				append_dev(div0, select);
+				append_dev(select, option0);
+				append_dev(select, option1);
+				append_dev(select, option2);
+				select_option(select, /*anthropicModel*/ ctx[1], true);
+				insert_dev(target, t12, anchor);
+				insert_dev(target, div6, anchor);
+				append_dev(div6, div3);
+				append_dev(div3, h20);
+				append_dev(div3, t14);
 				append_dev(div3, div2);
 
 				for (let i = 0; i < each_blocks.length; i += 1) {
@@ -9219,58 +9362,49 @@ var app = (function () {
 					}
 				}
 
-				insert_dev(target, t8, anchor);
-				insert_dev(target, div6, anchor);
+				append_dev(div6, t15);
 				append_dev(div6, div5);
-				append_dev(div5, input);
+				append_dev(div5, h21);
+				append_dev(div5, t17);
+				append_dev(div5, div4);
+				key_block.m(div4, null);
+				insert_dev(target, t18, anchor);
+				insert_dev(target, div8, anchor);
+				append_dev(div8, div7);
+				append_dev(div7, input);
 				set_input_value(input, /*model*/ ctx[0]);
-				append_dev(div5, t9);
-				append_dev(div5, button);
+				append_dev(div7, t19);
+				append_dev(div7, button3);
 
 				if (!mounted) {
 					dispose = [
-						listen_dev(input, "input", /*input_input_handler*/ ctx[8]),
-						listen_dev(button, "click", /*convert*/ ctx[5], false, false, false, false)
+						listen_dev(button0, "click", /*padToMatch*/ ctx[7], false, false, false, false),
+						listen_dev(button1, "click", /*removeAllSkipLayers*/ ctx[8], false, false, false, false),
+						listen_dev(button2, "click", /*matchByName*/ ctx[9], false, false, false, false),
+						listen_dev(select, "change", /*select_change_handler*/ ctx[10]),
+						listen_dev(input, "input", /*input_input_handler*/ ctx[13]),
+						listen_dev(button3, "click", /*convert*/ ctx[6], false, false, false, false)
 					];
 
 					mounted = true;
 				}
 			},
 			p: function update(ctx, [dirty]) {
-				if (dirty & /*String, jaxFields*/ 2) {
-					each_value_1 = ensure_array_like_dev(/*jaxFields*/ ctx[1]);
-					let i;
-
-					for (i = 0; i < each_value_1.length; i += 1) {
-						const child_ctx = get_each_context_1(ctx, each_value_1, i);
-
-						if (each_blocks_1[i]) {
-							each_blocks_1[i].p(child_ctx, dirty);
-						} else {
-							each_blocks_1[i] = create_each_block_1(child_ctx);
-							each_blocks_1[i].c();
-							each_blocks_1[i].m(div0, null);
-						}
-					}
-
-					for (; i < each_blocks_1.length; i += 1) {
-						each_blocks_1[i].d(1);
-					}
-
-					each_blocks_1.length = each_value_1.length;
+				if (dirty & /*anthropicModel*/ 2) {
+					select_option(select, /*anthropicModel*/ ctx[1]);
 				}
 
-				if (dirty & /*addSkipLayer, removeSkipLayer, torchFields, String*/ 28) {
-					each_value = ensure_array_like_dev(/*torchFields*/ ctx[2]);
+				if (dirty & /*String, jaxFields*/ 4) {
+					each_value_1 = ensure_array_like_dev(/*jaxFields*/ ctx[2]);
 					let i;
 
-					for (i = 0; i < each_value.length; i += 1) {
-						const child_ctx = get_each_context(ctx, each_value, i);
+					for (i = 0; i < each_value_1.length; i += 1) {
+						const child_ctx = get_each_context_1(ctx, each_value_1, i);
 
 						if (each_blocks[i]) {
 							each_blocks[i].p(child_ctx, dirty);
 						} else {
-							each_blocks[i] = create_each_block(child_ctx);
+							each_blocks[i] = create_each_block_1(child_ctx);
 							each_blocks[i].c();
 							each_blocks[i].m(div2, null);
 						}
@@ -9280,7 +9414,16 @@ var app = (function () {
 						each_blocks[i].d(1);
 					}
 
-					each_blocks.length = each_value.length;
+					each_blocks.length = each_value_1.length;
+				}
+
+				if (dirty & /*torchFields*/ 8 && safe_not_equal(previous_key, previous_key = /*torchFields*/ ctx[3])) {
+					key_block.d(1);
+					key_block = create_key_block(ctx);
+					key_block.c();
+					key_block.m(div4, null);
+				} else {
+					key_block.p(ctx, dirty);
 				}
 
 				if (dirty & /*model*/ 1 && input.value !== /*model*/ ctx[0]) {
@@ -9294,13 +9437,15 @@ var app = (function () {
 					detach_dev(t0);
 					detach_dev(h1);
 					detach_dev(t2);
-					detach_dev(div4);
-					detach_dev(t8);
+					detach_dev(div1);
+					detach_dev(t12);
 					detach_dev(div6);
+					detach_dev(t18);
+					detach_dev(div8);
 				}
 
-				destroy_each(each_blocks_1, detaching);
 				destroy_each(each_blocks, detaching);
+				key_block.d(detaching);
 				mounted = false;
 				run_all(dispose);
 			}
@@ -9317,13 +9462,6 @@ var app = (function () {
 		return block;
 	}
 
-	function swap(a, b, array) {
-		const temp = array[a];
-		array[a] = array[b];
-		array[b] = temp;
-		return array;
-	}
-
 	function checkFields(jaxFields, torchFields) {
 		if (jaxFields.length > torchFields.length) {
 			return {
@@ -9359,6 +9497,7 @@ var app = (function () {
 		let { $$slots: slots = {}, $$scope } = $$props;
 		validate_slots('App', slots, []);
 		let model = "model.eqx";
+		let anthropicModel = "haiku";
 
 		const Toast = Swal.mixin({
 			toast: true,
@@ -9374,13 +9513,12 @@ var app = (function () {
 
 		let jaxFields = [];
 		let torchFields = [];
-		let torchSortable;
 
 		onMount(async () => {
 			let req = await fetch("/startup/getJaxFields");
-			$$invalidate(1, jaxFields = await req.json());
+			$$invalidate(2, jaxFields = await req.json());
 			req = await fetch("/startup/getTorchFields");
-			$$invalidate(2, torchFields = await req.json());
+			$$invalidate(3, torchFields = await req.json());
 
 			setTimeout(
 				() => {
@@ -9388,10 +9526,17 @@ var app = (function () {
 				},
 				100
 			);
+
+			setTimeout(
+				() => {
+					onEnd();
+				},
+				500
+			);
 		});
 
 		function initSortable() {
-			torchSortable = new Sortable(document.getElementById("torch-fields"),
+			new Sortable(document.getElementById("torch-fields"),
 			{
 					animation: 150,
 					multiDrag: true,
@@ -9466,74 +9611,67 @@ var app = (function () {
 		}
 
 		function onEnd() {
-			var _a, _b, _c, _d, _e, _f, _g;
-			const updatedFields = fetchJaxAndTorchFields();
-
-			if (updatedFields.error) {
-				Toast.fire({
-					icon: "error",
-					title: updatedFields.error
-				});
-
-				return;
-			}
-
-			for (let i = 0; i < updatedFields.jaxFields.length; i++) {
-				let jaxField = updatedFields.jaxFields[i];
-				let torchField = updatedFields.torchFields[i];
-				if (torchField === undefined) continue;
+			setTimeout(
+				() => {
+					var _a, _b, _c, _d;
+					const updatedFields = fetchJaxAndTorchFields();
 
-				if (torchField.skip === true) {
-					(_a = document.getElementById("jax-" + i)) === null || _a === void 0
-					? void 0
-					: _a.classList.remove("bg-error");
+					if (updatedFields.error) {
+						Toast.fire({
+							icon: "error",
+							title: updatedFields.error
+						});
 
-					(_b = document.getElementById("torch-" + i)) === null || _b === void 0
-					? void 0
-					: _b.classList.remove("bg-error");
+						return;
+					}
 
-					continue;
-				}
+					for (let i = 0; i < updatedFields.jaxFields.length; i++) {
+						let jaxField = updatedFields.jaxFields[i];
+						let torchField = updatedFields.torchFields[i];
+						if (torchField === undefined) continue;
 
-				let jaxShape = jaxField.shape;
-				let torchShape = torchField.shape;
+						if (torchField.skip === true) {
+							(_a = document.getElementById("jax-" + i)) === null || _a === void 0
+							? void 0
+							: _a.classList.remove("bg-error");
 
-				//@ts-ignore
-				let jaxShapeProduct = jaxShape.reduce((a, b) => a * b, 1);
+							continue;
+						}
 
-				//@ts-ignore
-				let torchShapeProduct = torchShape.reduce((a, b) => a * b, 1);
+						let jaxShape = jaxField.shape;
+						let torchShape = torchField.shape;
 
-				if (jaxShapeProduct !== torchShapeProduct) {
-					(_c = document.getElementById("jax-" + i)) === null || _c === void 0
-					? void 0
-					: _c.classList.add("bg-error");
+						//@ts-ignore
+						let jaxShapeProduct = jaxShape.reduce((a, b) => a * b, 1);
 
-					(_d = document.getElementById("torch-" + i)) === null || _d === void 0
-					? void 0
-					: _d.classList.add("bg-error");
-				} else {
-					(_e = document.getElementById("jax-" + i)) === null || _e === void 0
-					? void 0
-					: _e.classList.remove("bg-error");
+						//@ts-ignore
+						let torchShapeProduct = torchShape.reduce((a, b) => a * b, 1);
 
-					(_f = document.getElementById("torch-" + i)) === null || _f === void 0
-					? void 0
-					: _f.classList.remove("bg-error");
-				}
-			}
+						if (jaxShapeProduct !== torchShapeProduct) {
+							(_b = document.getElementById("jax-" + i)) === null || _b === void 0
+							? void 0
+							: _b.classList.add("bg-error");
+						} else {
+							(_c = document.getElementById("jax-" + i)) === null || _c === void 0
+							? void 0
+							: _c.classList.remove("bg-error");
+						}
+					}
 
-			if (updatedFields.torchFields.length > updatedFields.jaxFields.length) {
-				for (let i = updatedFields.jaxFields.length; i < updatedFields.torchFields.length; i++) {
-					(_g = document.getElementById("torch-" + i)) === null || _g === void 0
-					? void 0
-					: _g.classList.remove("bg-error");
-				}
-			}
+					if (updatedFields.torchFields.length > updatedFields.jaxFields.length) {
+						for (let i = updatedFields.jaxFields.length; i < updatedFields.torchFields.length; i++) {
+							(_d = document.getElementById("torch-" + i)) === null || _d === void 0
+							? void 0
+							: _d.classList.remove("bg-error");
+						}
+					}
+				},
+				100
+			);
 		}
 
 		function removeSkipLayer(index) {
-			$$invalidate(2, torchFields = torchFields.toSpliced(index, 1));
+			$$invalidate(3, torchFields = torchFields.toSpliced(index, 1));
 
 			setTimeout(
 				() => {
@@ -9544,14 +9682,21 @@ var app = (function () {
 		}
 
 		function addSkipLayer(index) {
+			let fields = fetchJaxAndTorchFields();
+
+			if (fields.error) {
+				Toast.fire({ icon: "error", text: fields.error });
+				return;
+			}
+
 			const newField = {
 				skip: true,
-				shape: [],
-				path: "",
-				type: ""
+				shape: [0],
+				path: "SKIP",
+				type: "SKIP"
 			};
 
-			$$invalidate(2, torchFields = torchFields.toSpliced(index, 0, newField));
+			$$invalidate(3, torchFields = fields.torchFields.toSpliced(index, 0, newField));
 
 			setTimeout(
 				() => {
@@ -9604,12 +9749,241 @@ var app = (function () {
 			}
 		}
 
+		function padToMatch() {
+			let fields = fetchJaxAndTorchFields();
+
+			if (fields.error) {
+				Toast.fire({ icon: "error", text: fields.error });
+				return;
+			}
+
+			if (fields.torchFields.length < fields.jaxFields.length) {
+				let toAdd = fields.jaxFields.length - fields.torchFields.length;
+
+				for (let i = 0; i < toAdd; i++) {
+					setTimeout(
+						() => {
+							console.log("adding skip at ", i);
+							addSkipLayer(fields.jaxFields.length + i);
+						},
+						100
+					);
+				}
+			}
+		}
+
+		function removeAllSkipLayers() {
+			let fields = fetchJaxAndTorchFields();
+
+			if (fields.error) {
+				Toast.fire({ icon: "error", text: fields.error });
+				return;
+			}
+
+			let filteredFields = [];
+
+			for (let i = 0; i < fields.torchFields.length; i++) {
+				if (fields.torchFields[i].skip === false) {
+					filteredFields.push(fields.torchFields[i]);
+				}
+			}
+
+			$$invalidate(3, torchFields = filteredFields);
+		}
+
+		async function matchByName() {
+			let fields = fetchJaxAndTorchFields();
+
+			if (fields.error) {
+				Toast.fire({ icon: "error", text: fields.error });
+				return;
+			}
+
+			if (fields.jaxFields.length !== fields.torchFields.length) {
+				Toast.fire({
+					icon: "error",
+					text: "PyTree and State Dict have diffent lengths. Make sure to pad first!"
+				});
+
+				return;
+			}
+
+			Toast.fire({
+				icon: "info",
+				title: "Matching by name...",
+				text: "This can take a while! Hold tight."
+			});
+
+			let content = `
+You will get two lists of strings. These strings are fields of a JAX and PyTorch model.
+For example:
+--JAX START--
+.layers[0].weight
+.layers[1].weight
+.layers[2].weight
+.layers[3].weight
+.layers[4].weight
+--JAX END--
+
+--PYTORCH START--
+layers.0.weight
+layers.1.weight
+layers.4.weight
+layers.2.weight
+layers.3.weight
+--PYTORCH END--
+
+As you can see, the order doesn't match. This means, you should look at the PyTorch fields and
+rearrange them, such that they match the JAX model. In the above example, the expected return value
+is:
+--PYTORCH START--
+layers.0.weight
+layers.1.weight
+layers.2.weight
+layers.3.weight
+layers.4.weight
+--PYTORCH END--
+
+Here's another example:
+--JAX START--
+.conv1.weight
+.bn1.weight
+.bn1.bias
+.bn1.state_index.init[0]
+.bn1.state_index.init[1]
+--JAX END--
+
+--PYTORCH START--
+bn1.running_mean
+bn1.running_var
+conv1.weight
+bn1.weight
+bn1.bias
+--PYTORCH END--
+
+The expected return value in this case is:
+
+--PYTORCH START--
+conv1.weight
+bn1.weight
+bn1.bias
+bn1.running_mean
+bn1.running_var
+--PYTORCH END--
+
+
+Sometimes, there are so-called "skip-layers" in the PyTorch model. Those can be put anywhere, preferably to
+the end, because your priority is to match those fields that can be matched first! Here's an example:
+
+--JAX START--
+.layers[0].weight
+.layers[1].weight
+.layers[2].weight
+.layers[3].weight
+.layers[4].weight
+--JAX END--
+
+--PYTORCH START--
+layers.0.weight
+SKIP
+layers.3.weight
+layers.2.weight
+layers.1.weight
+--PYTORCH START--
+
+This should return
+
+--PYTORCH START--
+layers.0.weight
+layers.1.weight
+layers.2.weight
+layers.3.weight
+SKIP
+--PYTORCH START--
+
+
+It's not always 100% which belongs to which. Use your best judgement. Start your response with
+--PYTORCH START-- and end it with --PYTORCH END--.
+
+
+Here's your input:
+--JAX START--
+        `;
+
+			for (let i = 0; i < fields.jaxFields.length; i++) {
+				content += fields.jaxFields[i].path + "\n";
+			}
+
+			content += "--JAX END--\n";
+			content += "\n";
+			content += "--PYTORCH START--\n";
+
+			for (let i = 0; i < fields.torchFields.length; i++) {
+				content += fields.torchFields[i].path + "\n";
+			}
+
+			content += "--PYTORCH END--";
+			console.log(content);
+
+			let req = await fetch("/anthropic", {
+				method: "POST",
+				headers: { "Content-Type": "application/json" },
+				body: JSON.stringify({ content, model: anthropicModel })
+			});
+
+			let res = await req.json();
+
+			if (res.error) {
+				Toast.fire({ icon: "error", text: res.error });
+				return;
+			}
+
+			console.log(res);
+			let responseContent = res.content;
+			let lines = responseContent.split("\n");
+			console.log(lines);
+			let rearrangedTorchFields = [];
+
+			for (let i = 0; i < lines.length; i++) {
+				let matchingTorchField = fields.torchFields.find(field => field.path === lines[i]);
+
+				if (matchingTorchField) {
+					rearrangedTorchFields.push(matchingTorchField);
+				}
+			}
+
+			if (fields.torchFields.length !== rearrangedTorchFields.length) {
+				Toast.fire({
+					icon: "error",
+					text: "Some fields are missing in the response. Try a different model instead."
+				});
+
+				return;
+			}
+
+			console.log("rearrangedTorchFields", rearrangedTorchFields);
+
+			setTimeout(
+				() => {
+					$$invalidate(3, torchFields = rearrangedTorchFields);
+					onEnd();
+					Toast.fire({ icon: "success", title: "Success" });
+				},
+				500
+			);
+		}
+
 		const writable_props = [];
 
 		Object.keys($$props).forEach(key => {
 			if (!~writable_props.indexOf(key) && key.slice(0, 2) !== '$$' && key !== 'slot') console_1.warn(`<App> was created with unknown prop '${key}'`);
 		});
 
+		function select_change_handler() {
+			anthropicModel = select_value(this);
+			$$invalidate(1, anthropicModel);
+		}
+
 		const click_handler = i => {
 			removeSkipLayer(i);
 		};
@@ -9628,25 +10002,27 @@ var app = (function () {
 			onMount,
 			Swal,
 			model,
+			anthropicModel,
 			Toast,
 			jaxFields,
 			torchFields,
-			torchSortable,
 			initSortable,
-			swap,
 			fetchJaxAndTorchFields,
 			onEnd,
 			checkFields,
 			removeSkipLayer,
 			addSkipLayer,
-			convert
+			convert,
+			padToMatch,
+			removeAllSkipLayers,
+			matchByName
 		});
 
 		$$self.$inject_state = $$props => {
 			if ('model' in $$props) $$invalidate(0, model = $$props.model);
-			if ('jaxFields' in $$props) $$invalidate(1, jaxFields = $$props.jaxFields);
-			if ('torchFields' in $$props) $$invalidate(2, torchFields = $$props.torchFields);
-			if ('torchSortable' in $$props) torchSortable = $$props.torchSortable;
+			if ('anthropicModel' in $$props) $$invalidate(1, anthropicModel = $$props.anthropicModel);
+			if ('jaxFields' in $$props) $$invalidate(2, jaxFields = $$props.jaxFields);
+			if ('torchFields' in $$props) $$invalidate(3, torchFields = $$props.torchFields);
 		};
 
 		if ($$props && "$$inject" in $$props) {
@@ -9655,11 +10031,16 @@ var app = (function () {
 
 		return [
 			model,
+			anthropicModel,
 			jaxFields,
 			torchFields,
 			removeSkipLayer,
 			addSkipLayer,
 			convert,
+			padToMatch,
+			removeAllSkipLayers,
+			matchByName,
+			select_change_handler,
 			click_handler,
 			click_handler_1,
 			input_input_handler