JS-Torch - biblioteca JavaScript semelhante ao PyTorch

• Desenvolvida do zero, seguindo a sintaxe do PyTorch
• Inclui camadas e funções de deep learning, mecanismo de diferenciação automática e um objeto Tensor completo com rastreamento de gradientes
• Operações de Tensor implementadas: Add, Subtract, Multiply, Divide, Matrix Multiply, Power, Square Root, Exponentiate, Log, Sum, Mean, Variance, Transpose, At, MaskedFill, Reshape
• Camadas de deep learning implementadas: nn.Linear, nn.MultiHeadSelfAttention, nn.FullyConnected, nn.Block, nn.Embedding, nn.PositionalEmbedding, nn.ReLU, nn.Softmax, nn.Dropout, nn.LayerNorm, nn.CrossEntropyLoss

[eduardoleao052]

Obrigado por compartilhar!