thinc 9.0.0.dev4__tar.gz → 9.0.0.dev5__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 (191) hide show
  1. {thinc-9.0.0.dev4/thinc.egg-info → thinc-9.0.0.dev5}/PKG-INFO +1 -1
  2. thinc-9.0.0.dev5/thinc/about.py +2 -0
  3. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/backends/cupy_ops.py +2 -2
  4. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/compat.py +3 -0
  5. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/shims/torchscript.py +1 -1
  6. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_basic_tagger.py +0 -1
  7. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5/thinc.egg-info}/PKG-INFO +1 -1
  8. thinc-9.0.0.dev4/thinc/about.py +0 -2
  9. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/LICENSE +0 -0
  10. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/MANIFEST.in +0 -0
  11. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/README.md +0 -0
  12. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/pyproject.toml +0 -0
  13. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/setup.cfg +0 -0
  14. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/setup.py +0 -0
  15. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/__init__.pxd +0 -0
  16. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/__init__.py +0 -0
  17. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/api.py +0 -0
  18. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/backends/__init__.pxd +0 -0
  19. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/backends/__init__.py +0 -0
  20. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/backends/_cupy_allocators.py +0 -0
  21. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/backends/_custom_kernels.cu +0 -0
  22. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/backends/_custom_kernels.py +0 -0
  23. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/backends/_murmur3.cu +0 -0
  24. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/backends/_param_server.py +0 -0
  25. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/backends/cblas.pxd +0 -0
  26. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/backends/cblas.pyx +0 -0
  27. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/backends/cpu_kernels.hh +0 -0
  28. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/backends/mps_ops.py +0 -0
  29. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/backends/numpy_ops.pxd +0 -0
  30. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/backends/numpy_ops.pyx +0 -0
  31. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/backends/ops.py +0 -0
  32. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/config.py +0 -0
  33. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/initializers.py +0 -0
  34. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/__init__.py +0 -0
  35. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/add.py +0 -0
  36. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/array_getitem.py +0 -0
  37. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/bidirectional.py +0 -0
  38. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/cauchysimilarity.py +0 -0
  39. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/chain.py +0 -0
  40. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/clipped_linear.py +0 -0
  41. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/clone.py +0 -0
  42. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/concatenate.py +0 -0
  43. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/dish.py +0 -0
  44. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/dropout.py +0 -0
  45. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/embed.py +0 -0
  46. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/expand_window.py +0 -0
  47. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/gelu.py +0 -0
  48. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/hard_swish.py +0 -0
  49. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/hard_swish_mobilenet.py +0 -0
  50. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/hashembed.py +0 -0
  51. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/layernorm.py +0 -0
  52. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/linear.py +0 -0
  53. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/list2array.py +0 -0
  54. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/list2padded.py +0 -0
  55. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/list2ragged.py +0 -0
  56. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/logistic.py +0 -0
  57. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/lstm.py +0 -0
  58. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/map_list.py +0 -0
  59. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/maxout.py +0 -0
  60. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/mish.py +0 -0
  61. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/multisoftmax.py +0 -0
  62. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/mxnetwrapper.py +0 -0
  63. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/noop.py +0 -0
  64. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/padded2list.py +0 -0
  65. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/parametricattention.py +0 -0
  66. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/parametricattention_v2.py +0 -0
  67. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/premap_ids.pyx +0 -0
  68. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/pytorchwrapper.py +0 -0
  69. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/ragged2list.py +0 -0
  70. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/reduce_first.py +0 -0
  71. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/reduce_last.py +0 -0
  72. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/reduce_max.py +0 -0
  73. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/reduce_mean.py +0 -0
  74. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/reduce_sum.py +0 -0
  75. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/relu.py +0 -0
  76. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/remap_ids.py +0 -0
  77. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/residual.py +0 -0
  78. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/resizable.py +0 -0
  79. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/siamese.py +0 -0
  80. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/sigmoid.py +0 -0
  81. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/sigmoid_activation.py +0 -0
  82. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/softmax.py +0 -0
  83. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/softmax_activation.py +0 -0
  84. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/sparselinear.pyx +0 -0
  85. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/strings2arrays.py +0 -0
  86. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/swish.py +0 -0
  87. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/tensorflowwrapper.py +0 -0
  88. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/torchscriptwrapper.py +0 -0
  89. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/tuplify.py +0 -0
  90. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/uniqued.py +0 -0
  91. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/with_array.py +0 -0
  92. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/with_array2d.py +0 -0
  93. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/with_cpu.py +0 -0
  94. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/with_debug.py +0 -0
  95. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/with_flatten.py +0 -0
  96. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/with_flatten_v2.py +0 -0
  97. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/with_getitem.py +0 -0
  98. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/with_list.py +0 -0
  99. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/with_nvtx_range.py +0 -0
  100. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/with_padded.py +0 -0
  101. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/with_ragged.py +0 -0
  102. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/with_reshape.py +0 -0
  103. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/layers/with_signpost_interval.py +0 -0
  104. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/loss.py +0 -0
  105. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/model.py +0 -0
  106. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/mypy.py +0 -0
  107. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/optimizers.py +0 -0
  108. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/py.typed +0 -0
  109. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/schedules.py +0 -0
  110. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/shims/__init__.py +0 -0
  111. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/shims/mxnet.py +0 -0
  112. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/shims/pytorch.py +0 -0
  113. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/shims/pytorch_grad_scaler.py +0 -0
  114. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/shims/shim.py +0 -0
  115. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/shims/tensorflow.py +0 -0
  116. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/__init__.py +0 -0
  117. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/backends/__init__.py +0 -0
  118. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/backends/test_mem.py +0 -0
  119. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/backends/test_ops.py +0 -0
  120. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/conftest.py +0 -0
  121. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/enable_mxnet.py +0 -0
  122. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/enable_tensorflow.py +0 -0
  123. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/__init__.py +0 -0
  124. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_combinators.py +0 -0
  125. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_feed_forward.py +0 -0
  126. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_hash_embed.py +0 -0
  127. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_layers_api.py +0 -0
  128. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_linear.py +0 -0
  129. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_lstm.py +0 -0
  130. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_mappers.py +0 -0
  131. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_mnist.py +0 -0
  132. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_mxnet_wrapper.py +0 -0
  133. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_parametric_attention_v2.py +0 -0
  134. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_pytorch_wrapper.py +0 -0
  135. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_reduce.py +0 -0
  136. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_resizable.py +0 -0
  137. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_shim.py +0 -0
  138. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_softmax.py +0 -0
  139. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_sparse_linear.py +0 -0
  140. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_tensorflow_wrapper.py +0 -0
  141. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_torchscriptwrapper.py +0 -0
  142. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_transforms.py +0 -0
  143. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_uniqued.py +0 -0
  144. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_with_debug.py +0 -0
  145. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_with_flatten.py +0 -0
  146. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/layers/test_with_transforms.py +0 -0
  147. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/model/__init__.py +0 -0
  148. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/model/test_model.py +0 -0
  149. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/model/test_validation.py +0 -0
  150. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/mypy/__init__.py +0 -0
  151. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/mypy/configs/mypy-default.ini +0 -0
  152. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/mypy/configs/mypy-plugin.ini +0 -0
  153. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/mypy/modules/__init__.py +0 -0
  154. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/mypy/modules/fail_no_plugin.py +0 -0
  155. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/mypy/modules/fail_plugin.py +0 -0
  156. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/mypy/modules/success_no_plugin.py +0 -0
  157. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/mypy/modules/success_plugin.py +0 -0
  158. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/mypy/outputs/fail-no-plugin.txt +0 -0
  159. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/mypy/outputs/fail-plugin.txt +0 -0
  160. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/mypy/outputs/success-no-plugin.txt +0 -0
  161. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/mypy/outputs/success-plugin.txt +0 -0
  162. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/mypy/test_mypy.py +0 -0
  163. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/regression/__init__.py +0 -0
  164. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/regression/issue519/__init__.py +0 -0
  165. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/regression/issue519/program.py +0 -0
  166. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/regression/issue519/test_issue519.py +0 -0
  167. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/regression/test_issue208.py +0 -0
  168. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/regression/test_issue564.py +0 -0
  169. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/shims/__init__.py +0 -0
  170. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/shims/test_pytorch_grad_scaler.py +0 -0
  171. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/strategies.py +0 -0
  172. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/test_config.py +0 -0
  173. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/test_examples.py +0 -0
  174. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/test_import__all__.py +0 -0
  175. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/test_indexing.py +0 -0
  176. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/test_initializers.py +0 -0
  177. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/test_loss.py +0 -0
  178. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/test_optimizers.py +0 -0
  179. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/test_schedules.py +0 -0
  180. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/test_serialize.py +0 -0
  181. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/test_types.py +0 -0
  182. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/test_util.py +0 -0
  183. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/tests/util.py +0 -0
  184. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/types.py +0 -0
  185. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc/util.py +0 -0
  186. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc.egg-info/SOURCES.txt +0 -0
  187. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc.egg-info/dependency_links.txt +0 -0
  188. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc.egg-info/entry_points.txt +0 -0
  189. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc.egg-info/not-zip-safe +0 -0
  190. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc.egg-info/requires.txt +0 -0
  191. {thinc-9.0.0.dev4 → thinc-9.0.0.dev5}/thinc.egg-info/top_level.txt +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: thinc
3
- Version: 9.0.0.dev4
3
+ Version: 9.0.0.dev5
4
4
  Summary: A refreshing functional take on deep learning, compatible with your favorite libraries
5
5
  Home-page: https://github.com/explosion/thinc
6
6
  Author: Explosion
@@ -0,0 +1,2 @@
1
+ __version__ = "9.0.0.dev5"
2
+ __release__ = True
@@ -1,7 +1,7 @@
1
1
  import numpy
2
2
 
3
3
  from .. import registry
4
- from ..compat import cupy, cupyx
4
+ from ..compat import cublas, cupy, cupyx
5
5
  from ..types import DeviceTypes
6
6
  from ..util import (
7
7
  is_cupy_array,
@@ -257,7 +257,7 @@ class CupyOps(Ops):
257
257
  # implementation.
258
258
  def frobenius_norm(X):
259
259
  X_vec = X.reshape(-1)
260
- return cupy.cublas.nrm2(X_vec)
260
+ return cublas.nrm2(X_vec)
261
261
 
262
262
  grad_norm = cupy.maximum(frobenius_norm(gradient), 1e-12)
263
263
  gradient *= cupy.minimum(threshold, grad_norm) / grad_norm
@@ -4,9 +4,11 @@ from packaging.version import Version
4
4
 
5
5
  try: # pragma: no cover
6
6
  import cupy
7
+ import cupy.cublas
7
8
  import cupyx
8
9
 
9
10
  has_cupy = True
11
+ cublas = cupy.cublas
10
12
  cupy_version = Version(cupy.__version__)
11
13
  try:
12
14
  cupy.cuda.runtime.getDeviceCount()
@@ -20,6 +22,7 @@ try: # pragma: no cover
20
22
  else:
21
23
  cupy_from_dlpack = cupy.fromDlpack
22
24
  except (ImportError, AttributeError):
25
+ cublas = None
23
26
  cupy = None
24
27
  cupyx = None
25
28
  cupy_version = Version("0.0.0")
@@ -30,7 +30,7 @@ class TorchScriptShim(PyTorchShim):
30
30
 
31
31
  def __init__(
32
32
  self,
33
- model: Optional["torch.ScriptModule"],
33
+ model: Optional["torch.jit.ScriptModule"],
34
34
  config=None,
35
35
  optimizer: Any = None,
36
36
  mixed_precision: bool = False,
@@ -60,7 +60,6 @@ def get_shuffled_batches(Xs, Ys, batch_size):
60
60
  yield list(batch_X), list(batch_Y)
61
61
 
62
62
 
63
- @pytest.mark.slow
64
63
  @pytest.mark.parametrize(
65
64
  ("depth", "width", "vector_width", "nb_epoch"), [(2, 32, 16, 5)]
66
65
  )
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: thinc
3
- Version: 9.0.0.dev4
3
+ Version: 9.0.0.dev5
4
4
  Summary: A refreshing functional take on deep learning, compatible with your favorite libraries
5
5
  Home-page: https://github.com/explosion/thinc
6
6
  Author: Explosion
@@ -1,2 +0,0 @@
1
- __version__ = "9.0.0.dev4"
2
- __release__ = True
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes