nmn 0.1.9__py3-none-any.whl → 0.1.10__py3-none-any.whl

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 (5) hide show
  1. nmn/nnx/squashers/softermax.py +6 -1
  2. {nmn-0.1.9.dist-info → nmn-0.1.10.dist-info}/METADATA +1 -1
  3. {nmn-0.1.9.dist-info → nmn-0.1.10.dist-info}/RECORD +5 -5
  4. {nmn-0.1.9.dist-info → nmn-0.1.10.dist-info}/WHEEL +0 -0
  5. {nmn-0.1.9.dist-info → nmn-0.1.10.dist-info}/licenses/LICENSE +0 -0
nmn/nnx/squashers/softermax.py CHANGED
@@ -1,7 +1,12 @@
1
+ from functools import partial
2
+ from typing import Optional
3
+
4
+ import jax
1
5
  import jax.numpy as jnp
2
6
  from jax import Array
3
- from typing import Optional
4
7
 
8
+
9
+ @partial(jax.jit, static_argnames=("n", "axis"))
5
10
  def softermax(
6
11
  x: Array,
7
12
  n: float = 1.0,
{nmn-0.1.9.dist-info → nmn-0.1.10.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: nmn
3
- Version: 0.1.9
3
+ Version: 0.1.10
4
4
  Summary: a neuron that matter
5
5
  Project-URL: Homepage, https://github.com/mlnomadpy/nmn
6
6
  Project-URL: Bug Tracker, https://github.com/mlnomadpy/my_package/issues
{nmn-0.1.9.dist-info → nmn-0.1.10.dist-info}/RECORD RENAMED
@@ -11,11 +11,11 @@ nmn/nnx/loss/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
11
11
  nmn/nnx/squashers/__init__.py,sha256=zXYPa3yzqMXxkIPvNHiaV6pcZRDOdVrzaVdYVDGALTY,180
12
12
  nmn/nnx/squashers/soft_tanh.py,sha256=WSJkxD6L9WU1eqPwsK2AW4V6OJbw5pSWYjKwkiWtLdo,812
13
13
  nmn/nnx/squashers/softer_sigmoid.py,sha256=vE6IWorZdBb2cww6fskARnwzdjTcWB2kKohuaJWVGNs,845
14
- nmn/nnx/squashers/softermax.py,sha256=NfxEDbogLUysyTvtVCTpDt27PplYvKRQLTZbYCL-Wfg,1226
14
+ nmn/nnx/squashers/softermax.py,sha256=ggg0mHMFyk7b5xs31o-inNvWDzEvghD6YO3mtPlnkW4,1318
15
15
  nmn/tf/nmn.py,sha256=A-K65z9_aN62tAy12b0553nXxrzOofK1umGMRGJYjqw,6036
16
16
  nmn/torch/conv.py,sha256=g5YxStk1p85WkvfecqbzRZaWaAJahOSArpMcqxWAWKc,83413
17
17
  nmn/torch/nmn.py,sha256=8K0S3nwpGprT7apbCqpaYpKpxq8F8g8EL8PHIezgMCY,4658
18
- nmn-0.1.9.dist-info/METADATA,sha256=HHYtGZLUmGNgYhmxH6d5P8RWh7QytALKOhlfpMwa9fg,8800
19
- nmn-0.1.9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
20
- nmn-0.1.9.dist-info/licenses/LICENSE,sha256=kbZSd5WewnN2PSjvAC6DprP7pXx6NUNsnltmU2Mz1yA,34519
21
- nmn-0.1.9.dist-info/RECORD,,
18
+ nmn-0.1.10.dist-info/METADATA,sha256=o-wLjeO-n2h56-cvw-AqrRiio5UFaerm58w03XkdHQY,8801
19
+ nmn-0.1.10.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
20
+ nmn-0.1.10.dist-info/licenses/LICENSE,sha256=kbZSd5WewnN2PSjvAC6DprP7pXx6NUNsnltmU2Mz1yA,34519
21
+ nmn-0.1.10.dist-info/RECORD,,
{nmn-0.1.9.dist-info → nmn-0.1.10.dist-info}/WHEEL RENAMED
File without changes