Add Transformer

frost-beta · May 4, 2024 · e53f4be · e53f4be
1 parent 560f28c
commit e53f4be
Show file tree

Hide file tree

Showing 2 changed files with 421 additions and 0 deletions.
diff --git a/lib/nn/index.ts b/lib/nn/index.ts
@@ -10,6 +10,7 @@ export * from './layers/pooling';
 export * from './layers/positional-encoding';
 export * from './layers/quantized';
 export * from './layers/recurrent';
+export * from './layers/transformer';
 export * from './layers/upsample';
 export * from './utils';
 export * as init from './init';