• PyTorch

  • Referenced in 199 articles [sw20939]
  • PyTorch python package: Tensors and Dynamic neural networks in Python with strong GPU acceleration. PyTorch...
  • CycleGAN

  • Referenced in 32 articles [sw28302]
  • CycleGAN and pix2pix in PyTorch. We provide PyTorch implementations for both unpaired and paired image ... Park, and supported by Tongzhou Wang. This PyTorch implementation produces results comparable to or better...
  • pix2pix

  • Referenced in 32 articles [sw28303]
  • CycleGAN and pix2pix in PyTorch. We provide PyTorch implementations for both unpaired and paired image ... Park, and supported by Tongzhou Wang. This PyTorch implementation produces results comparable to or better...
  • torchdiffeq

  • Referenced in 24 articles [sw35082]
  • torchdiffeq: PyTorch Implementation of Differentiable ODE Solvers. This library provides ordinary differential equation (ODE) solvers ... implemented in PyTorch. Backpropagation through all solvers is supported using the adjoint method. For usage ... solvers are implemented in PyTorch, algorithms in this repository are fully supported...
  • AllenNLP

  • Referenced in 13 articles [sw26526]
  • open-source NLP research library, built on PyTorch. AllenNLP makes it easy to design...
  • Transformers

  • Referenced in 7 articles [sw30739]
  • Transformers (formerly known as pytorch-transformers and pytorch-pretrained-bert) provides state ... deep interoperability between TensorFlow 2.0 and PyTorch...
  • Pyro

  • Referenced in 11 articles [sw27079]
  • written in Python and supported by PyTorch on the backend. Pyro enables flexible and expressive...
  • TensorLy

  • Referenced in 7 articles [sw27020]
  • seamlessly perform computation with NumPy, MXNet, PyTorch, TensorFlow or CuPy, and run methods at scale...
  • RLlib

  • Referenced in 7 articles [sw31153]
  • RLlib natively supports TensorFlow, TensorFlow Eager, and PyTorch, but most of its internals are framework...
  • GPyTorch

  • Referenced in 7 articles [sw35483]
  • scalable GP inference via BBMM, built on PyTorch...
  • PyTorch-BigGraph

  • Referenced in 3 articles [sw34086]
  • PyTorch-BigGraph: A Large-scale Graph Embedding System. Graph embedding methods produce unsupervised node features ... capability of existing embedding systems. We present PyTorch-BigGraph (PBG), an embedding system that incorporates...
  • Torchmeta

  • Referenced in 3 articles [sw30740]
  • Torchmeta: A Meta-Learning library for PyTorch. The constant introduction of standardized benchmarks ... Torchmeta, a library built on top of PyTorch that enables seamless and consistent evaluation ... abstraction. It also features some extensions for PyTorch to simplify the development of models compatible...
  • Kornia

  • Referenced in 3 articles [sw31197]
  • Open Source Differentiable Computer Vision Library for PyTorch. This work presents Kornia -- an open source ... generic computer vision problems. The package uses PyTorch as its main backend both for efficiency...
  • Captum

  • Referenced in 3 articles [sw35075]
  • unified and generic model interpretability library for PyTorch. In this paper we introduce a novel ... unified, open-source model interpretability library for PyTorch [12]. The library contains generic implementations...
  • rTorch

  • Referenced in 2 articles [sw33942]
  • package rTorch: R Bindings to ’PyTorch’. ’R’ implementation and interface of the Machine Learning platform ... PyTorch developed in ’Python’. It requires a ’conda’ environment with ’torch’ and ’torchvision ... provide ’PyTorch’ functions, methods and classes. The key object in ’PyTorch’ is the tensor which...
  • tntorch

  • Referenced in 2 articles [sw35476]
  • tntorch - Tensor Network Learning with PyTorch. This is a PyTorch-powered library for tensor modeling...
  • CogDL

  • Referenced in 2 articles [sw37740]
  • CogDL are developed on top of PyTorch, and users can leverage the advantages of PyTorch...
  • SchNetPack

  • Referenced in 3 articles [sw25772]
  • molecule and material datasets. Based upon the PyTorch deep learning framework, SchNetPack allows to efficiently...
  • KeOps

  • Referenced in 3 articles [sw30747]
  • used through Matlab, NumPy or PyTorch backends. It is perfectly suited to the computation...
  • AutoGraph

  • Referenced in 3 articles [sw30957]
  • learning, imperative style libraries like Autograd and PyTorch are easy to write, but suffer from...