Add layer norm usage for Transformers.js (#11)

- Add layer norm usage for Transformers.js (0351d4f49f2b7816a4cbbe488dc3d4aaab8296c9)


Co-authored-by: Joshua <[email protected]>

README.md

@@ -2730,7 +2730,7 @@ The model natively supports scaling of the sequence length past 2048 tokens. To
 ### Transformers.js
 
 ```js
-import { pipeline } from '@xenova/transformers';
+import { pipeline, layer_norm } from '@xenova/transformers';
 
 // Create a feature extraction pipeline
 const extractor = await pipeline('feature-extraction', 'nomic-ai/nomic-embed-text-v1.5', {
@@ -2745,8 +2745,10 @@ let embeddings = await extractor(texts, { pooling: 'mean' });
 console.log(embeddings); // Tensor of shape [2, 768]
 
 const matryoshka_dim = 512;
-embeddings = embeddings.slice(null, [0, matryoshka_dim]).normalize(2, -1);
-console.log(embeddings); // Tensor of shape [2, 512]
+embeddings = layer_norm(embeddings, [embeddings.dims[1]])
+    .slice(null, [0, matryoshka_dim])
+    .normalize(2, -1);
+console.log(embeddings.tolist());
 ```
 
 # Join the Nomic Community