netcl 0.2.2__tar.gz → 0.2.3__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (115) hide show
  1. {netcl-0.2.2/netcl.egg-info → netcl-0.2.3}/PKG-INFO +1 -1
  2. {netcl-0.2.2 → netcl-0.2.3}/netcl/autograd/__init__.py +14 -0
  3. {netcl-0.2.2 → netcl-0.2.3}/netcl/autograd/ops.py +1253 -0
  4. {netcl-0.2.2 → netcl-0.2.3}/netcl/nn/__init__.py +11 -2
  5. {netcl-0.2.2 → netcl-0.2.3}/netcl/nn/layers.py +142 -1
  6. {netcl-0.2.2 → netcl-0.2.3}/netcl/nn/loss.py +58 -0
  7. {netcl-0.2.2 → netcl-0.2.3/netcl.egg-info}/PKG-INFO +1 -1
  8. {netcl-0.2.2 → netcl-0.2.3}/pyproject.toml +1 -1
  9. {netcl-0.2.2 → netcl-0.2.3}/LICENSE +0 -0
  10. {netcl-0.2.2 → netcl-0.2.3}/MANIFEST.in +0 -0
  11. {netcl-0.2.2 → netcl-0.2.3}/README.md +0 -0
  12. {netcl-0.2.2 → netcl-0.2.3}/README_PACKAGE.md +0 -0
  13. {netcl-0.2.2 → netcl-0.2.3}/netcl/__init__.py +0 -0
  14. {netcl-0.2.2 → netcl-0.2.3}/netcl/amp.py +0 -0
  15. {netcl-0.2.2 → netcl-0.2.3}/netcl/autograd/compiler.py +0 -0
  16. {netcl-0.2.2 → netcl-0.2.3}/netcl/autograd/debug.py +0 -0
  17. {netcl-0.2.2 → netcl-0.2.3}/netcl/autograd/engine.py +0 -0
  18. {netcl-0.2.2 → netcl-0.2.3}/netcl/autograd/forward_compiler.py +0 -0
  19. {netcl-0.2.2 → netcl-0.2.3}/netcl/autograd/graph.py +0 -0
  20. {netcl-0.2.2 → netcl-0.2.3}/netcl/autograd/training_compiler.py +0 -0
  21. {netcl-0.2.2 → netcl-0.2.3}/netcl/core/__init__.py +0 -0
  22. {netcl-0.2.2 → netcl-0.2.3}/netcl/core/backend/__init__.py +0 -0
  23. {netcl-0.2.2 → netcl-0.2.3}/netcl/core/backend/base.py +0 -0
  24. {netcl-0.2.2 → netcl-0.2.3}/netcl/core/backend/cpu.py +0 -0
  25. {netcl-0.2.2 → netcl-0.2.3}/netcl/core/backend/opencl.py +0 -0
  26. {netcl-0.2.2 → netcl-0.2.3}/netcl/core/capabilities.py +0 -0
  27. {netcl-0.2.2 → netcl-0.2.3}/netcl/core/device.py +0 -0
  28. {netcl-0.2.2 → netcl-0.2.3}/netcl/core/device_profile.py +0 -0
  29. {netcl-0.2.2 → netcl-0.2.3}/netcl/core/kernel_selector.py +0 -0
  30. {netcl-0.2.2 → netcl-0.2.3}/netcl/core/kernels/__init__.py +0 -0
  31. {netcl-0.2.2 → netcl-0.2.3}/netcl/core/kernels/primitives.py +0 -0
  32. {netcl-0.2.2 → netcl-0.2.3}/netcl/core/memory.py +0 -0
  33. {netcl-0.2.2 → netcl-0.2.3}/netcl/core/parameter.py +0 -0
  34. {netcl-0.2.2 → netcl-0.2.3}/netcl/core/shape.py +0 -0
  35. {netcl-0.2.2 → netcl-0.2.3}/netcl/core/tensor.py +0 -0
  36. {netcl-0.2.2 → netcl-0.2.3}/netcl/data/augment.py +0 -0
  37. {netcl-0.2.2 → netcl-0.2.3}/netcl/data/augment_gpu.py +0 -0
  38. {netcl-0.2.2 → netcl-0.2.3}/netcl/data/dataloader.py +0 -0
  39. {netcl-0.2.2 → netcl-0.2.3}/netcl/data/filters.py +0 -0
  40. {netcl-0.2.2 → netcl-0.2.3}/netcl/data/shared_batch.py +0 -0
  41. {netcl-0.2.2 → netcl-0.2.3}/netcl/distributed/__init__.py +0 -0
  42. {netcl-0.2.2 → netcl-0.2.3}/netcl/distributed/collectives.py +0 -0
  43. {netcl-0.2.2 → netcl-0.2.3}/netcl/distributed/data_parallel.py +0 -0
  44. {netcl-0.2.2 → netcl-0.2.3}/netcl/distributed/device_manager.py +0 -0
  45. {netcl-0.2.2 → netcl-0.2.3}/netcl/distributed/trainer.py +0 -0
  46. {netcl-0.2.2 → netcl-0.2.3}/netcl/io/__init__.py +0 -0
  47. {netcl-0.2.2 → netcl-0.2.3}/netcl/io/checkpoint.py +0 -0
  48. {netcl-0.2.2 → netcl-0.2.3}/netcl/io/serialization.py +0 -0
  49. {netcl-0.2.2 → netcl-0.2.3}/netcl/nn/batchnorm.py +0 -0
  50. {netcl-0.2.2 → netcl-0.2.3}/netcl/nn/decorators.py +0 -0
  51. {netcl-0.2.2 → netcl-0.2.3}/netcl/nn/factory.py +0 -0
  52. {netcl-0.2.2 → netcl-0.2.3}/netcl/nn/functional.py +0 -0
  53. {netcl-0.2.2 → netcl-0.2.3}/netcl/nn/groupnorm.py +0 -0
  54. {netcl-0.2.2 → netcl-0.2.3}/netcl/nn/init.py +0 -0
  55. {netcl-0.2.2 → netcl-0.2.3}/netcl/nn/layernorm.py +0 -0
  56. {netcl-0.2.2 → netcl-0.2.3}/netcl/nn/modules.py +0 -0
  57. {netcl-0.2.2 → netcl-0.2.3}/netcl/nn/padding.py +0 -0
  58. {netcl-0.2.2 → netcl-0.2.3}/netcl/nn/pooling.py +0 -0
  59. {netcl-0.2.2 → netcl-0.2.3}/netcl/nn/residual.py +0 -0
  60. {netcl-0.2.2 → netcl-0.2.3}/netcl/nn/resnet.py +0 -0
  61. {netcl-0.2.2 → netcl-0.2.3}/netcl/nn/simple.py +0 -0
  62. {netcl-0.2.2 → netcl-0.2.3}/netcl/ops/__init__.py +0 -0
  63. {netcl-0.2.2 → netcl-0.2.3}/netcl/ops/bmm.py +0 -0
  64. {netcl-0.2.2 → netcl-0.2.3}/netcl/ops/broadcast.py +0 -0
  65. {netcl-0.2.2 → netcl-0.2.3}/netcl/ops/conv2d.py +0 -0
  66. {netcl-0.2.2 → netcl-0.2.3}/netcl/ops/conv2d_cpu.py +0 -0
  67. {netcl-0.2.2 → netcl-0.2.3}/netcl/ops/conv2d_optimized.py +0 -0
  68. {netcl-0.2.2 → netcl-0.2.3}/netcl/ops/conv2d_planner.py +0 -0
  69. {netcl-0.2.2 → netcl-0.2.3}/netcl/ops/conv_transpose2d.py +0 -0
  70. {netcl-0.2.2 → netcl-0.2.3}/netcl/ops/depthwise_conv2d.py +0 -0
  71. {netcl-0.2.2 → netcl-0.2.3}/netcl/ops/elementwise.py +0 -0
  72. {netcl-0.2.2 → netcl-0.2.3}/netcl/ops/elementwise_optimized.py +0 -0
  73. {netcl-0.2.2 → netcl-0.2.3}/netcl/ops/fused_ops.py +0 -0
  74. {netcl-0.2.2 → netcl-0.2.3}/netcl/ops/im2col.py +0 -0
  75. {netcl-0.2.2 → netcl-0.2.3}/netcl/ops/jit_fusion.py +0 -0
  76. {netcl-0.2.2 → netcl-0.2.3}/netcl/ops/matmul.py +0 -0
  77. {netcl-0.2.2 → netcl-0.2.3}/netcl/ops/matmul_optimized.py +0 -0
  78. {netcl-0.2.2 → netcl-0.2.3}/netcl/ops/permute.py +0 -0
  79. {netcl-0.2.2 → netcl-0.2.3}/netcl/ops/reduction.py +0 -0
  80. {netcl-0.2.2 → netcl-0.2.3}/netcl/ops/softmax.py +0 -0
  81. {netcl-0.2.2 → netcl-0.2.3}/netcl/ops/softmax_fp16.py +0 -0
  82. {netcl-0.2.2 → netcl-0.2.3}/netcl/ops/softmax_fused.py +0 -0
  83. {netcl-0.2.2 → netcl-0.2.3}/netcl/ops/transpose.py +0 -0
  84. {netcl-0.2.2 → netcl-0.2.3}/netcl/ops/winograd_fused.py +0 -0
  85. {netcl-0.2.2 → netcl-0.2.3}/netcl/optim/__init__.py +0 -0
  86. {netcl-0.2.2 → netcl-0.2.3}/netcl/optim/_utils.py +0 -0
  87. {netcl-0.2.2 → netcl-0.2.3}/netcl/optim/adam.py +0 -0
  88. {netcl-0.2.2 → netcl-0.2.3}/netcl/optim/adamw.py +0 -0
  89. {netcl-0.2.2 → netcl-0.2.3}/netcl/optim/amp.py +0 -0
  90. {netcl-0.2.2 → netcl-0.2.3}/netcl/optim/clip.py +0 -0
  91. {netcl-0.2.2 → netcl-0.2.3}/netcl/optim/lr_plateau.py +0 -0
  92. {netcl-0.2.2 → netcl-0.2.3}/netcl/optim/lr_scheduler.py +0 -0
  93. {netcl-0.2.2 → netcl-0.2.3}/netcl/optim/momentum.py +0 -0
  94. {netcl-0.2.2 → netcl-0.2.3}/netcl/optim/rmsprop.py +0 -0
  95. {netcl-0.2.2 → netcl-0.2.3}/netcl/optim/sgd.py +0 -0
  96. {netcl-0.2.2 → netcl-0.2.3}/netcl/profiling/__init__.py +0 -0
  97. {netcl-0.2.2 → netcl-0.2.3}/netcl/profiling/autotuner.py +0 -0
  98. {netcl-0.2.2 → netcl-0.2.3}/netcl/profiling/timing.py +0 -0
  99. {netcl-0.2.2 → netcl-0.2.3}/netcl/runtime/__init__.py +0 -0
  100. {netcl-0.2.2 → netcl-0.2.3}/netcl/runtime/cache.py +0 -0
  101. {netcl-0.2.2 → netcl-0.2.3}/netcl/runtime/capture.py +0 -0
  102. {netcl-0.2.2 → netcl-0.2.3}/netcl/runtime/graph.py +0 -0
  103. {netcl-0.2.2 → netcl-0.2.3}/netcl/runtime/graph_fusion.py +0 -0
  104. {netcl-0.2.2 → netcl-0.2.3}/netcl/runtime/perf.py +0 -0
  105. {netcl-0.2.2 → netcl-0.2.3}/netcl/runtime/scheduler.py +0 -0
  106. {netcl-0.2.2 → netcl-0.2.3}/netcl/trainer/__init__.py +0 -0
  107. {netcl-0.2.2 → netcl-0.2.3}/netcl/trainer/trainer.py +0 -0
  108. {netcl-0.2.2 → netcl-0.2.3}/netcl/utils/__init__.py +0 -0
  109. {netcl-0.2.2 → netcl-0.2.3}/netcl/utils/data.py +0 -0
  110. {netcl-0.2.2 → netcl-0.2.3}/netcl/utils/progress.py +0 -0
  111. {netcl-0.2.2 → netcl-0.2.3}/netcl.egg-info/SOURCES.txt +0 -0
  112. {netcl-0.2.2 → netcl-0.2.3}/netcl.egg-info/dependency_links.txt +0 -0
  113. {netcl-0.2.2 → netcl-0.2.3}/netcl.egg-info/requires.txt +0 -0
  114. {netcl-0.2.2 → netcl-0.2.3}/netcl.egg-info/top_level.txt +0 -0
  115. {netcl-0.2.2 → netcl-0.2.3}/setup.cfg +0 -0
{netcl-0.2.2/netcl.egg-info → netcl-0.2.3}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.2
2
2
  Name: netcl
3
- Version: 0.2.2
3
+ Version: 0.2.3
4
4
  Summary: PyOpenCL-based deep learning playground with autograd, kernels, and high-level APIs.
5
5
  Author: netcl contributors
6
6
  License: MIT
{netcl-0.2.2 → netcl-0.2.3}/netcl/autograd/__init__.py RENAMED
@@ -82,6 +82,13 @@ from .ops import (
82
82
  permute,
83
83
  bmm,
84
84
  softmax,
85
+ triplet_loss,
86
+ info_nce_loss,
87
+ pairwise_squared_distances,
88
+ batch_hard_triplet_loss,
89
+ l2_normalize,
90
+ contrastive_loss,
91
+ batch_all_triplet_loss,
85
92
  )
86
93
 
87
94
  __all__ = [
@@ -164,6 +171,13 @@ __all__ = [
164
171
  "permute",
165
172
  "bmm",
166
173
  "softmax",
174
+ "triplet_loss",
175
+ "info_nce_loss",
176
+ "pairwise_squared_distances",
177
+ "batch_hard_triplet_loss",
178
+ "l2_normalize",
179
+ "contrastive_loss",
180
+ "batch_all_triplet_loss",
167
181
  "debug_tape",
168
182
  "set_current_tape",
169
183
  "get_current_tape",