Skip to content

torch-directml 0.2.2.dev240614

Pre-release
Pre-release
Compare
Choose a tag to compare
@joshjkim joshjkim released this 15 Jun 01:16
· 17 commits to master since this release
torch-directml-0.2.2.dev240614
372a622
This commit was created on GitHub.com and signed with GitHub’s verified signature.
GPG key ID: B5690EEEBB952194
Learn about vigilant mode.

Preview build of torch-directml built on June 14, 2024.

The Python packages are available as a PyPI release. To download the appropriate python package automatically, simply pip install torch-directml.

What’s New

  • Support for PyTorch 2.3.1 and torchvision 0.18.1
  • Operator support for softplus, amax, linspace, vector_norm, native_dropout
  • Performance improvements for linear operator
  • Initial support for SparseTensor via dense tensors
  • Added sample and support for OpenAI Whisper model
  • LLM sample code refactor and cleanup for improved maintainability
  • Added Generator register and support for random seed for torch PrivateUse1 backend

Bug Fixes

  • Fix for layer norm producing incorrect result with non-contiguous input #588
  • Issues addressed with new operators and PrivateUse1 random seed support #592 #590 #587 #586
Assets 12
Loading