@imaduddin/fastbrain 2.0.0 → 2.0.2

package/dist/Embedding/index.js

@@ -1,5 +1,8 @@
 import { GetVectorsForFile, SaveFileVector, initRandomVector } from '../fileManager.js';
 import { GPUManager } from '@imaduddin/webgpu-manager';
+import { fileURLToPath } from 'node:url';
+import path, { join } from 'node:path';
+const __dirname = path.dirname(fileURLToPath(import.meta.url));
 export class Embedding {
     device;
     embeddingSize;
@@ -38,7 +41,7 @@ export class Embedding {
             }
         }
         this.gpu = new GPUManager(device, {
-            shaderDir: "./shaders"
+            shaderDir: join(__dirname, "shaders/")
         });
     }
     async forward(input, encoderShared) {

package/dist/Gelu/index.js

@@ -1,4 +1,7 @@
 import { GPUManager } from "@imaduddin/webgpu-manager";
+import { fileURLToPath } from 'node:url';
+import path, { join } from 'node:path';
+const __dirname = path.dirname(fileURLToPath(import.meta.url));
 export class Gelu {
     device;
     hiddenSize;
@@ -13,7 +16,7 @@ export class Gelu {
         this.device = device;
         this.hiddenSize = hiddenSize;
         this.gpu = new GPUManager(device, {
-            shaderDir: "./shaders/relu"
+            shaderDir: join(__dirname, "./shaders/relu")
         });
     }
     async forward(input, encoderShared) {

package/dist/Linear/index.js

@@ -1,5 +1,8 @@
 import { GPUManager } from "@imaduddin/webgpu-manager";
 import { GetVectorsForFile, SaveFileVector, initRandomVector, getParameter } from '../fileManager.js';
+import { fileURLToPath } from 'node:url';
+import path, { join } from 'node:path';
+const __dirname = path.dirname(fileURLToPath(import.meta.url));
 export class Linear {
     device;
     embeddingSize;
@@ -49,7 +52,7 @@ export class Linear {
             }
         }
         this.gpu = new GPUManager(device, {
-            shaderDir: "./shaders"
+            shaderDir: join(__dirname, "shaders/")
         });
     }
     getParameter() {

package/dist/RMSNorm/index.js

@@ -1,4 +1,7 @@
 import { GPUManager } from "@imaduddin/webgpu-manager";
+import { fileURLToPath } from 'node:url';
+import path, { join } from 'node:path';
+const __dirname = path.dirname(fileURLToPath(import.meta.url));
 export class RMSNorm {
     device;
     embeddingSize;
@@ -14,7 +17,7 @@ export class RMSNorm {
         this.device = device;
         this.embeddingSize = embeddingSize;
         this.gpu = new GPUManager(device, {
-            shaderDir: "./shaders/rmsnorm"
+            shaderDir: join(__dirname, "./shaders/rmsnorm")
         });
     }
     async forward(input, encoderShared) {

package/dist/SelfAttention/index.js

@@ -1,6 +1,8 @@
-import { join } from "node:path";
 import { Linear } from "../Linear/index.js";
 import { GPUManager } from "@imaduddin/webgpu-manager";
+import { fileURLToPath } from 'node:url';
+import path, { join } from 'node:path';
+const __dirname = path.dirname(fileURLToPath(import.meta.url));
 export class SelfAttention {
     device;
     embeddingSize;
@@ -47,7 +49,7 @@ export class SelfAttention {
             filename: [join(configs.dirname, "LINEAR_KEY_WEIGHTS.bin"), join(configs.dirname, "LINEAR_KEY_BIAS.bin")]
         });
         this.gpu = new GPUManager(device, {
-            shaderDir: "./shaders"
+            shaderDir: join(__dirname, "shaders/")
         });
     }
     async softmaxBackward(weights, dWeigths, output, tokenCount, enocder) {

package/dist/Softmax/index.js

@@ -1,5 +1,8 @@
 import { GPUManager } from "@imaduddin/webgpu-manager";
 import { isBufferOrNumber } from '../fileManager.js';
+import { fileURLToPath } from 'node:url';
+import path, { join } from 'node:path';
+const __dirname = path.dirname(fileURLToPath(import.meta.url));
 export class Softmax {
     device;
     VocabSize;
@@ -16,7 +19,7 @@ export class Softmax {
         this.device = device;
         this.VocabSize = VocabSize;
         this.gpu = new GPUManager(device, {
-            shaderDir: './shaders/softmax'
+            shaderDir: join(__dirname, './shaders/softmax')
         });
     }
     async forward(logit, encoderShared) {

package/dist/residualConnection/index.js

@@ -1,4 +1,7 @@
 import { GPUManager } from "@imaduddin/webgpu-manager";
+import { fileURLToPath } from 'node:url';
+import path, { join } from 'node:path';
+const __dirname = path.dirname(fileURLToPath(import.meta.url));
 export class residualConnection {
     device;
     output;
@@ -7,7 +10,7 @@ export class residualConnection {
     constructor(device) {
         this.device = device;
         this.gpu = new GPUManager(device, {
-            shaderDir: "./shaders"
+            shaderDir: join(__dirname, "./shaders")
         });
     }
     async add(a, b) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@imaduddin/fastbrain",
-  "version": "2.0.0",
+  "version": "2.0.2",
   "description": "Raw-JAVASCRIPT Based Matirx and AI LLM Package called fastBrain.",
   "keywords": [
     "ai",
@@ -12,6 +12,12 @@
     "Language",
     "Model"
   ],
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js"
+    }
+  },
   "license": "MIT",
   "author": "mineproness",
   "type": "module",

package/src/Embedding/index.ts

@@ -1,5 +1,10 @@
 import { GetVectorsForFile, SaveFileVector, initRandomVector } from '../fileManager.js'
 import { Buffer, GPUManager } from '@imaduddin/webgpu-manager';
+import { readFileSync } from 'node:fs';
+import { fileURLToPath } from 'node:url';
+import path, { join } from 'node:path';
+
+const __dirname = path.dirname(fileURLToPath(import.meta.url));
 
 
 export class Embedding {
@@ -31,7 +36,7 @@ export class Embedding {
             }
         }
         this.gpu = new GPUManager(device, {
-            shaderDir: "./shaders"
+            shaderDir: join(__dirname, "shaders/")
         })
     }
     async forward(input: number[], encoderShared: GPUCommandEncoder | undefined) {

package/src/Gelu/index.ts

@@ -1,6 +1,10 @@
 import { Buffer, GPUManager } from "@imaduddin/webgpu-manager";
 
+import { readFileSync } from 'node:fs';
+import { fileURLToPath } from 'node:url';
+import path, { join } from 'node:path';
 
+const __dirname = path.dirname(fileURLToPath(import.meta.url));
 export class Gelu {
     input: Buffer | undefined;
     output: Buffer | undefined;
@@ -11,7 +15,7 @@ export class Gelu {
     gpu: GPUManager | undefined;
     constructor(public device: GPUDevice, public hiddenSize: number) {
         this.gpu = new GPUManager(device, {
-            shaderDir: "./shaders/relu"
+            shaderDir: join(__dirname,"./shaders/relu")
         })
     }
     async forward(input: Buffer, encoderShared: GPUCommandEncoder | undefined) {

package/src/Linear/index.ts

@@ -1,6 +1,10 @@
 import { GPUManager, Buffer } from "@imaduddin/webgpu-manager";
 import { GetVectorsForFile, SaveFileVector, initRandomVector, getParameter } from '../fileManager.js'
+import { readFileSync } from 'node:fs';
+import { fileURLToPath } from 'node:url';
+import path, { join } from 'node:path';
 
+const __dirname = path.dirname(fileURLToPath(import.meta.url));
 
 export class Linear {
     bias: Float32Array | undefined;
@@ -42,7 +46,7 @@ export class Linear {
             }
         }
         this.gpu = new GPUManager(device, {
-            shaderDir: "./shaders"
+            shaderDir: join(__dirname, "shaders/")
         })
     }
     getParameter(){

package/src/RMSNorm/index.ts

@@ -1,5 +1,9 @@
 import { GPUManager, Buffer } from "@imaduddin/webgpu-manager";
+import { readFileSync } from 'node:fs';
+import { fileURLToPath } from 'node:url';
+import path, { join } from 'node:path';
 
+const __dirname = path.dirname(fileURLToPath(import.meta.url));
 
 export class RMSNorm {
     rms: Buffer | undefined
@@ -12,7 +16,7 @@ export class RMSNorm {
     gpu: GPUManager | undefined
     constructor(public device: GPUDevice, public embeddingSize: number) {
         this.gpu = new GPUManager(device, {
-            shaderDir: "./shaders/rmsnorm"
+            shaderDir: join(__dirname, "./shaders/rmsnorm")
         })
     }
     async forward(input: Buffer, encoderShared: GPUCommandEncoder | undefined) {

package/src/SelfAttention/index.ts

@@ -1,7 +1,11 @@
-import { join } from "node:path";
+
 import { Linear } from "../Linear/index.js";
 import { GPUManager, Buffer } from "@imaduddin/webgpu-manager";
+import { readFileSync } from 'node:fs';
+import { fileURLToPath } from 'node:url';
+import path, { join } from 'node:path';
 
+const __dirname = path.dirname(fileURLToPath(import.meta.url));
 export class SelfAttention {
     query: Linear | undefined
     value: Linear | undefined
@@ -44,7 +48,7 @@ export class SelfAttention {
             filename: [join(configs.dirname, "LINEAR_KEY_WEIGHTS.bin"), join(configs.dirname, "LINEAR_KEY_BIAS.bin")]
         })
         this.gpu = new GPUManager(device, {
-            shaderDir: "./shaders"
+            shaderDir: join(__dirname, "shaders/")
         })
     }
     async softmaxBackward(weights: Buffer, dWeigths: Buffer, output: Buffer, tokenCount: number, enocder: GPUCommandEncoder) {

package/src/Softmax/index.ts

@@ -1,6 +1,10 @@
 import { GPUManager, Buffer } from "@imaduddin/webgpu-manager";
 import { isBufferOrNumber } from '../fileManager.js'
+import { readFileSync } from 'node:fs';
+import { fileURLToPath } from 'node:url';
+import path, { join } from 'node:path';
 
+const __dirname = path.dirname(fileURLToPath(import.meta.url));
 export class Softmax {
     forwardShader: GPUComputePipeline | undefined;
     lossShader: GPUComputePipeline | undefined;
@@ -13,7 +17,7 @@ export class Softmax {
     targetBuffer: Buffer | undefined;
     constructor(public device: GPUDevice, public VocabSize: number) {
         this.gpu = new GPUManager(device, {
-            shaderDir: './shaders/softmax'
+            shaderDir: join(__dirname,'./shaders/softmax')
         })
     }
     async forward(logit: Buffer | number[], encoderShared: GPUCommandEncoder | undefined) {

package/src/residualConnection/index.ts

@@ -1,5 +1,9 @@
 import {Buffer,GPUManager} from "@imaduddin/webgpu-manager"
+import { readFileSync } from 'node:fs';
+import { fileURLToPath } from 'node:url';
+import path, { join } from 'node:path';
 
+const __dirname = path.dirname(fileURLToPath(import.meta.url));
 
 export class residualConnection{
     output: Buffer | undefined;
@@ -7,7 +11,7 @@ export class residualConnection{
     addShader: GPUComputePipeline | undefined;
     constructor(public device : GPUDevice){
         this.gpu = new GPUManager(device, {
-            shaderDir: "./shaders"
+            shaderDir: join(__dirname, "./shaders")
         })
     }
     async add(a: Buffer, b: Buffer){