JS-Torch - PyTorch に似た JavaScript ライブラリ
(github.com/eduardoleao052)- PyTorch の構文に従いながら、ゼロから新規開発
- ディープラーニングのレイヤーと関数、自動微分(Differentiation)エンジン、Gradient 追跡が可能なフル機能の Tensor オブジェクトを内蔵
- 実装済みの Tensor Operation: Add, Subtract, Multiply, Divide, Matrix Multiply, Power, Square Root, Exponentiate, Log, Sum, Mean, Variance, Transpose, At, MaskedFill, Reshape
- 実装済みのディープラーニング Layers: nn.Linear, nn.MultiHeadSelfAttention, nn.FullyConnected, nn.Block, nn.Embedding, nn.PositionalEmbedding, nn.ReLU, nn.Softmax, nn.Dropout, nn.LayerNorm, nn.CrossEntropyLoss
1件のコメント
投稿ありがとうございます!