@jax-js/jax 0.0.3 → 0.0.4

package/dist/{webgpu-fqhx41TC.cjs → webgpu-BVdMaO9T.cjs}

@@ -1,4 +1,4 @@
-const require_backend = require('./backend-D2C4MJRP.cjs');
+const require_backend = require('./backend-Ss1Mev_-.cjs');
 
 //#region src/backend/webgpu.ts
 /** Implementation of `Backend` that uses WebGPU in browsers. */
@@ -254,17 +254,23 @@ function pipelineSource(device, kernel) {
 			else if (op === require_backend.AluOp.Max) source = `max(${require_backend.strip1(a)}, ${require_backend.strip1(b)})`;
 			else if (op === require_backend.AluOp.Cmplt) source = `(${a} < ${b})`;
 			else if (op === require_backend.AluOp.Cmpne) source = `(${a} != ${b})`;
-		} else if (require_backend.AluGroup.Unary.has(op)) {
+		} else if (require_backend.AluGroup.Unary.has(op)) if (op === require_backend.AluOp.Reciprocal && src[0].op === require_backend.AluOp.Sqrt) {
+			const a = gen(src[0].src[0]);
+			source = `inverseSqrt(${a})`;
+		} else {
 			const a = gen(src[0]);
 			if (op === require_backend.AluOp.Sin) source = `sin(${a})`;
 			else if (op === require_backend.AluOp.Cos) source = `cos(${a})`;
+			else if (op === require_backend.AluOp.Asin) source = `asin(${a})`;
+			else if (op === require_backend.AluOp.Atan) source = `atan(${a})`;
 			else if (op === require_backend.AluOp.Exp) source = `exp(${a})`;
 			else if (op === require_backend.AluOp.Log) source = `log(${a})`;
 			else if (op === require_backend.AluOp.Sqrt) source = `sqrt(${a})`;
 			else if (op === require_backend.AluOp.Reciprocal) source = `(1.0 / ${a})`;
 			else if (op === require_backend.AluOp.Cast) source = `${dtypeToWgsl(dtype)}(${require_backend.strip1(a)})`;
 			else if (op === require_backend.AluOp.Bitcast) source = `bitcast<${dtypeToWgsl(dtype)}>(${require_backend.strip1(a)})`;
-		} else if (op === require_backend.AluOp.Where) source = `select(${require_backend.strip1(gen(src[2]))}, ${require_backend.strip1(gen(src[1]))}, ${require_backend.strip1(gen(src[0]))})`;
+		}
+		else if (op === require_backend.AluOp.Where) source = `select(${require_backend.strip1(gen(src[2]))}, ${require_backend.strip1(gen(src[1]))}, ${require_backend.strip1(gen(src[0]))})`;
 		else if (op === require_backend.AluOp.Threefry2x32) {
 			const x = gensym();
 			const [k0, k1, c0, c1] = src.map((x$1) => require_backend.strip1(gen(x$1)));

package/dist/{webgpu-CNg9JGva.js → webgpu-ow0Pn_6q.js}

@@ -1,4 +1,4 @@
-import { AluExp, AluGroup, AluOp, DEBUG, DType, Executable, FpHash, SlotError, UnsupportedOpError, findPow2, isFloatDtype, strip1, tuneWebgpu, union } from "./backend-BqDtPGaR.js";
+import { AluExp, AluGroup, AluOp, DEBUG, DType, Executable, FpHash, SlotError, UnsupportedOpError, findPow2, isFloatDtype, strip1, tuneWebgpu, union } from "./backend-EBRGmEYw.js";
 
 //#region src/backend/webgpu.ts
 /** Implementation of `Backend` that uses WebGPU in browsers. */
@@ -254,17 +254,23 @@ function pipelineSource(device, kernel) {
 			else if (op === AluOp.Max) source = `max(${strip1(a)}, ${strip1(b)})`;
 			else if (op === AluOp.Cmplt) source = `(${a} < ${b})`;
 			else if (op === AluOp.Cmpne) source = `(${a} != ${b})`;
-		} else if (AluGroup.Unary.has(op)) {
+		} else if (AluGroup.Unary.has(op)) if (op === AluOp.Reciprocal && src[0].op === AluOp.Sqrt) {
+			const a = gen(src[0].src[0]);
+			source = `inverseSqrt(${a})`;
+		} else {
 			const a = gen(src[0]);
 			if (op === AluOp.Sin) source = `sin(${a})`;
 			else if (op === AluOp.Cos) source = `cos(${a})`;
+			else if (op === AluOp.Asin) source = `asin(${a})`;
+			else if (op === AluOp.Atan) source = `atan(${a})`;
 			else if (op === AluOp.Exp) source = `exp(${a})`;
 			else if (op === AluOp.Log) source = `log(${a})`;
 			else if (op === AluOp.Sqrt) source = `sqrt(${a})`;
 			else if (op === AluOp.Reciprocal) source = `(1.0 / ${a})`;
 			else if (op === AluOp.Cast) source = `${dtypeToWgsl(dtype)}(${strip1(a)})`;
 			else if (op === AluOp.Bitcast) source = `bitcast<${dtypeToWgsl(dtype)}>(${strip1(a)})`;
-		} else if (op === AluOp.Where) source = `select(${strip1(gen(src[2]))}, ${strip1(gen(src[1]))}, ${strip1(gen(src[0]))})`;
+		}
+		else if (op === AluOp.Where) source = `select(${strip1(gen(src[2]))}, ${strip1(gen(src[1]))}, ${strip1(gen(src[0]))})`;
 		else if (op === AluOp.Threefry2x32) {
 			const x = gensym();
 			const [k0, k1, c0, c1] = src.map((x$1) => strip1(gen(x$1)));

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@jax-js/jax",
-  "version": "0.0.3",
+  "version": "0.0.4",
   "description": "Numerical computing and ML in the browser",
   "keywords": [
     "machine learning",
@@ -38,7 +38,7 @@
   "devDependencies": {
     "@eslint/js": "^9.31.0",
     "@types/debug": "^4.1.12",
-    "@vitest/browser": "^3.2.4",
+    "@vitest/browser-playwright": "^4.0.9",
     "@webgpu/types": "^0.1.64",
     "eslint": "^9.31.0",
     "eslint-plugin-import": "^2.32.0",
@@ -52,7 +52,7 @@
     "typedoc-theme-fresh": "^0.2.1",
     "typescript": "~5.9.3",
     "typescript-eslint": "^8.46.4",
-    "vitest": "^3.2.4"
+    "vitest": "^4.0.9"
   },
   "engines": {
     "pnpm": ">=10.0.0"
@@ -60,7 +60,18 @@
   "prettier": {
     "plugins": [
       "prettier-plugin-svelte"
-    ]
+    ],
+    "overrides": [
+      {
+        "files": [
+          "*.md"
+        ],
+        "options": {
+          "printWidth": 100
+        }
+      }
+    ],
+    "proseWrap": "always"
   },
   "scripts": {
     "build": "tsdown",