Facebook released PyTorch 1.4 today with upgrades to audio, vision, and text libraries; customization to the mobile version of its deep learning framework; and limited support of the Java programming language. PyTorch is one of the most popular machine learning frameworks used by researchers and developers today.
In the latest version, developers can use fine-grain customization of build scripts with PyTorch Mobile. This is a way to optimize library sizes and greatly reduce the on-device footprint, and in early tests was able to generate a version of MobileNetV2 edge model up to 50% smaller.
PyTorch 1.4 also includes a framework for distributed model parallel training and Java support for PyTorch inference based on the PyTorch Mobile for Android interface, but at launch the experimental feature is only available for Linux and for inference. PyTorch currently supports Python and C++.
PyTorch Mobile for iOS and Android devices first became available last fall as part of the release of PyTorch 1.3, with speed gains coming from quantization, Google TPU support, and a JIT compiler upgrade. Tools like Captum for explainable machine learning and CryptTen, a tool for encrypted machine learning, were also released last fall at the annual PyTorch developer conference.
With 1.4, domain libraries are also getting an upgrade, such as torchvision, which is getting ONNX support for all models, and torchscriptable support for all models to make them easier to use in non-Python environments.
The torchaudio library upgrade comes with a number of new filters and an interactive speech recognition in experimentation, while the torchtext library will gain access to the enwik9 unsupervised learning data set.
Additional PyTorch features were introduced last month, such as the image and video classification framework PyTorch Elastic.
VentureBeatVentureBeat's mission is to be a digital town square for technical decision-makers to gain knowledge about transformative technology and transact. Our site delivers essential information on data technologies and strategies to guide you as you lead your organizations. We invite you to become a member of our community, to access:
- up-to-date information on the subjects of interest to you
- our newsletters
- gated thought-leader content and discounted access to our prized events, such as Transform 2021: Learn More
- networking features, and more