jactorch package

Subpackages

• jactorch.cuda package
  • Submodules
  • jactorch.cuda.copy module
• jactorch.data package
  • Subpackages
  • Submodules
  • jactorch.data.dataset module
  • jactorch.data.layout module
• jactorch.functional package
  • Submodules
  • jactorch.functional.arith module
  • jactorch.functional.clustering module
  • jactorch.functional.grad module
  • jactorch.functional.indexing module
  • jactorch.functional.kernel module
  • jactorch.functional.linalg module
  • jactorch.functional.loglinear module
  • jactorch.functional.masking module
  • jactorch.functional.probability module
  • jactorch.functional.quantization module
  • jactorch.functional.range module
  • jactorch.functional.sampling module
  • jactorch.functional.shape module
• jactorch.graph package
  • Submodules
  • jactorch.graph.context module
  • jactorch.graph.nn_env module
  • jactorch.graph.parameter module
  • jactorch.graph.variable module
• jactorch.models package
  • Subpackages
• jactorch.nn package
  • Subpackages
  • Submodules
  • jactorch.nn.container module
  • jactorch.nn.embedding module
  • jactorch.nn.functional module
  • jactorch.nn.linear module
  • jactorch.nn.normalization module
  • jactorch.nn.probability module
  • jactorch.nn.quickaccess module
  • jactorch.nn.residual module
  • jactorch.nn.rnn_layers module
  • jactorch.nn.rnn_utils module
  • jactorch.nn.simple module
  • jactorch.nn.word_embedding module
• jactorch.optim package
  • Submodules
  • jactorch.optim.accum_grad module
  • jactorch.optim.custom_optimizer module
  • jactorch.optim.optimizer_group module
  • jactorch.optim.quickaccess module
  • jactorch.optim.weight_decay module
• jactorch.parallel package
  • Submodules
  • jactorch.parallel.comm module
  • jactorch.parallel.data_parallel module
  • jactorch.parallel.dict_gather module
  • jactorch.parallel.replicate module
  • jactorch.parallel.replication_callback module
  • jactorch.parallel.user_scattered module
• jactorch.quickstart package
  • Submodules
  • jactorch.quickstart.inference module
  • jactorch.quickstart.models module
  • jactorch.quickstart.train module
• jactorch.train package
  • Submodules
  • jactorch.train.env module
  • jactorch.train.monitor module
  • jactorch.train.tb module
  • jactorch.train.utils module
• jactorch.transforms package
  • Subpackages
• jactorch.utils package
  • Submodules
  • jactorch.utils.grad module
  • jactorch.utils.init module
  • jactorch.utils.meta module
  • jactorch.utils.unittest module
• jactorch.vision package
  • Submodules
  • jactorch.vision.conv module
  • jactorch.vision.geometry module
  • jactorch.vision.gradient module
  • jactorch.vision.morphology module
  • jactorch.vision.peak module
  • jactorch.vision.smooth module

Submodules

jactorch.cli module

jactorch.io module