JS-Torch - une bibliothèque JavaScript similaire à PyTorch

• Développée entièrement from scratch tout en suivant la syntaxe de PyTorch
• Intègre des layers et fonctions de deep learning, un moteur complet de différentiation automatique et un objet Tensor pleinement fonctionnel capable de suivre les gradients
• Opérations Tensor implémentées : Add, Subtract, Multiply, Divide, Matrix Multiply, Power, Square Root, Exponentiate, Log, Sum, Mean, Variance, Transpose, At, MaskedFill, Reshape
• Layers de deep learning implémentés : nn.Linear, nn.MultiHeadSelfAttention, nn.FullyConnected, nn.Block, nn.Embedding, nn.PositionalEmbedding, nn.ReLU, nn.Softmax, nn.Dropout, nn.LayerNorm, nn.CrossEntropyLoss

[eduardoleao052]

Merci pour le partage !