Skip to content

Releases: eduardoleao052/js-pytorch

js-pytorch 0.7.2

30 Oct 02:50
7a3212f
Compare
Choose a tag to compare

New features:

  • cdnjs tag available, for easier integration into web browsers and apps.
  • nn.MSELoss added, with Seyronh's help.
  • Fixed saving and loading models when using gpu kernels.

js-pytorch 0.5.0

20 Jul 22:37
Compare
Choose a tag to compare

In this version, 2 main features were added:

  • 1. GPU support
    • The back-end GPU kernels are managed by the gpu.js library.
    • To add a Tensor to the gpu, pass 'gpu' as the device argument.
    • To add a whole layer (nn.Linear, nn.Block, nn.MultiHeadSelfAttention) to the gpu, , pass 'gpu' as the device argument.
    • The syntax is just like the PyTorch counterpart.
  • 2. Saving and Loading models
    • The functions are torch.save and torch.load.
    • The functionality is explained on the README.md file.

js-pytorch 0.3.0

02 Apr 18:04
Compare
Choose a tag to compare

This new release contains the following features:

  • TypeScript support and source-code.
  • Developer tools, such as ESLint and prettier.
  • Automated unit tests with jest.
  • Benchmarking tools with tiny-bench.

Thanks to @pparke and @medic-code for the help with the implementations!

@eduardoleao052

js-pytorch 0.1.1

26 Mar 14:28
cbe6c37
Compare
Choose a tag to compare

Initial release. Package available on npm at https://www.npmjs.com/package/js-pytorch.