jinns 1.1.0__py3-none-any.whl → 1.2.0__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 (31) hide show
  1. jinns/data/_Batchs.py +4 -8
  2. jinns/data/_DataGenerators.py +532 -341
  3. jinns/loss/_DynamicLoss.py +150 -173
  4. jinns/loss/_DynamicLossAbstract.py +25 -73
  5. jinns/loss/_LossODE.py +3 -3
  6. jinns/loss/_LossPDE.py +27 -36
  7. jinns/loss/__init__.py +7 -6
  8. jinns/loss/_boundary_conditions.py +148 -279
  9. jinns/loss/_loss_utils.py +78 -56
  10. jinns/loss/_operators.py +441 -184
  11. jinns/plot/_plot.py +111 -98
  12. jinns/solver/_rar.py +102 -407
  13. jinns/solver/_solve.py +73 -38
  14. jinns/solver/_utils.py +122 -0
  15. jinns/utils/__init__.py +2 -0
  16. jinns/utils/_containers.py +3 -1
  17. jinns/utils/_hyperpinn.py +17 -7
  18. jinns/utils/_pinn.py +17 -27
  19. jinns/utils/_ppinn.py +227 -0
  20. jinns/utils/_save_load.py +13 -13
  21. jinns/utils/_spinn.py +24 -43
  22. jinns/utils/_types.py +1 -0
  23. jinns/utils/_utils.py +40 -12
  24. jinns-1.2.0.dist-info/METADATA +127 -0
  25. jinns-1.2.0.dist-info/RECORD +41 -0
  26. {jinns-1.1.0.dist-info → jinns-1.2.0.dist-info}/WHEEL +1 -1
  27. jinns-1.1.0.dist-info/METADATA +0 -85
  28. jinns-1.1.0.dist-info/RECORD +0 -39
  29. {jinns-1.1.0.dist-info → jinns-1.2.0.dist-info}/AUTHORS +0 -0
  30. {jinns-1.1.0.dist-info → jinns-1.2.0.dist-info}/LICENSE +0 -0
  31. {jinns-1.1.0.dist-info → jinns-1.2.0.dist-info}/top_level.txt +0 -0
jinns-1.2.0.dist-info/RECORD ADDED
@@ -0,0 +1,41 @@
1
+ jinns/__init__.py,sha256=5p7V5VJd7PXEINqhqS4mUsnQtXlyPwfctRhL4p0loFg,181
2
+ jinns/data/_Batchs.py,sha256=oc7-N1wEbsEvbe9fjVFKG2OPoZJVEjzPm8uj_icACf4,817
3
+ jinns/data/_DataGenerators.py,sha256=BWNF28jM7EN6wyJ3X2X1qjG4sjs6vHLgIVVY6aqhKS8,65099
4
+ jinns/data/__init__.py,sha256=TRCH0Z4-SQZ50MbSf46CUYWBkWVDmXCyez9T-EGiv_8,338
5
+ jinns/experimental/__init__.py,sha256=3jCIy2R2i_0Erwxg-HwISdH79Nt1XCXhS9yY1F5awiY,208
6
+ jinns/experimental/_diffrax_solver.py,sha256=upMr3kTTNrxEiSUO_oLvCXcjS9lPxSjvbB81h3qlhaU,6813
7
+ jinns/loss/_DynamicLoss.py,sha256=x1pJLZ_4P2iHhf5hjRhkWi1v5Q2dAH-v7Gctv4ax73E,25819
8
+ jinns/loss/_DynamicLossAbstract.py,sha256=bqmPxyrcvZh_dL74DTpj-TGiFxchvG8qC6KhuGeyOoA,12006
9
+ jinns/loss/_LossODE.py,sha256=lTJ3b2EnlYk1eAcldiVyFRh_XQrk83eYRQBuqWSiILg,23418
10
+ jinns/loss/_LossPDE.py,sha256=ufECwKdQXFEy8h5fdDA2uFTjiO_sVdqi9nfP7NTyu60,48775
11
+ jinns/loss/__init__.py,sha256=PRiJV9fd2GSwaCBVCPyh6pFc6pdA40jfb_T1YvO8ERc,712
12
+ jinns/loss/_boundary_conditions.py,sha256=KL_UUajQWLtRFqmBO_lfEcxrUsY-398ySTgpfaHdeYk,12277
13
+ jinns/loss/_loss_utils.py,sha256=RXu20VYeAelpuRIAJ67k3h7jdLsu2VIdjzmLVdiMjns,14223
14
+ jinns/loss/_loss_weights.py,sha256=F0Fgji2XpVk3pr9oIryGuXcG1FGQo4Dv6WFgze2BtA0,2201
15
+ jinns/loss/_operators.py,sha256=Hf5zWUQCGRSm1SS8f6IZLqC8bL7suC-R4ckXNbXqoAI,21790
16
+ jinns/parameters/__init__.py,sha256=1gxNLoAXUjhUzBWuh86YjU5pYy8SOboCs8TrKcU1wZc,158
17
+ jinns/parameters/_derivative_keys.py,sha256=UyEcgfNF1vwPcGWD2ShAZkZiq4thzRDm_OUJzOfjjiY,21909
18
+ jinns/parameters/_params.py,sha256=wK9ZSqoL9KnjOWqc_ZhJ09ffbsgeUEcttc1Rhme0lLk,3550
19
+ jinns/plot/__init__.py,sha256=Q279h5veYWNLQyttsC8_tDOToqUHh8WaRON90CiWXqk,81
20
+ jinns/plot/_plot.py,sha256=jS0fdI9VHCd6HL4K8ZzDjmMr10EJFqU6a0S4iWpEzFk,11433
21
+ jinns/solver/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
22
+ jinns/solver/_rar.py,sha256=BCvUdNLicdU1YizHKcSaJLmkUkCybyGmTRjHnjoiQKA,9730
23
+ jinns/solver/_solve.py,sha256=TR-RuH2tYIFq6YhbZ-Aa7r8OYy7CstjEYNm2RMI3xck,22641
24
+ jinns/solver/_utils.py,sha256=b2zYvwZY_fU0NMNWvUEMvHez9s7hwcxfpGzQlz5F6HA,5762
25
+ jinns/utils/__init__.py,sha256=nx7wi4RfgeHr5wLM_c1BAG8UQteTUTK6B0t9enmTn1E,243
26
+ jinns/utils/_containers.py,sha256=a7A-iUApnjc1YVc7bdt9tKUvHHPDOKMB9OfdrDZGWN8,1450
27
+ jinns/utils/_hyperpinn.py,sha256=hcdHRHXer03I3Twupok5moPR1vRGgPQUCjFHAZMCMdc,16872
28
+ jinns/utils/_pinn.py,sha256=8wDliUrDbBUKWllJgllapNX5BLLHMWF7Mz2CaEA-6vk,12667
29
+ jinns/utils/_ppinn.py,sha256=K4xuGaT-4XddIxsi1wOYO24G9l1RAY2m4bZqP7C3klI,8690
30
+ jinns/utils/_save_load.py,sha256=BngwUjkm5jibctnUCB8klMypYkfE6DKLaoWehwWnTNU,8585
31
+ jinns/utils/_spinn.py,sha256=tjly3qNZj9C2JntFKuwrLfFAbVfvy7Z5HPjj0ymfnLk,8337
32
+ jinns/utils/_types.py,sha256=EQXxUAipwy7lVk39hYw4NDf_a66OP01w_T3ziPITxpU,1901
33
+ jinns/utils/_utils.py,sha256=hoRcJqcTuQi_Ip40oI4EbxW46E1rp2C01_HfuCpwKRM,2932
34
+ jinns/validation/__init__.py,sha256=Jv58mzgC3F7cRfXA6caicL1t_U0UAhbwLrmMNVg6E7s,66
35
+ jinns/validation/_validation.py,sha256=bvqL2poTFJfn9lspWqMqXvQGcQIodKwKrC786QtEZ7A,4700
36
+ jinns-1.2.0.dist-info/AUTHORS,sha256=7NwCj9nU-HNG1asvy4qhQ2w7oZHrn-Lk5_wK_Ve7a3M,80
37
+ jinns-1.2.0.dist-info/LICENSE,sha256=BIAkGtXB59Q_BG8f6_OqtQ1BHPv60ggE9mpXJYz2dRM,11337
38
+ jinns-1.2.0.dist-info/METADATA,sha256=PxuzXslHG4YzOcuwDY3ulo9zJaF95ChjK9y584eMnN0,4662
39
+ jinns-1.2.0.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
40
+ jinns-1.2.0.dist-info/top_level.txt,sha256=RXbkr2hzy8WBE8aiRyrJYFqn3JeMJIhMdybLjjLTB9c,6
41
+ jinns-1.2.0.dist-info/RECORD,,
{jinns-1.1.0.dist-info → jinns-1.2.0.dist-info}/WHEEL RENAMED
@@ -1,5 +1,5 @@
1
1
  Wheel-Version: 1.0
2
- Generator: setuptools (75.1.0)
2
+ Generator: setuptools (75.6.0)
3
3
  Root-Is-Purelib: true
4
4
  Tag: py3-none-any
5
5
 
jinns-1.1.0.dist-info/METADATA DELETED
@@ -1,85 +0,0 @@
1
- Metadata-Version: 2.1
2
- Name: jinns
3
- Version: 1.1.0
4
- Summary: Physics Informed Neural Network with JAX
5
- Author-email: Hugo Gangloff <hugo.gangloff@inrae.fr>, Nicolas Jouvin <nicolas.jouvin@inrae.fr>
6
- Maintainer-email: Hugo Gangloff <hugo.gangloff@inrae.fr>, Nicolas Jouvin <nicolas.jouvin@inrae.fr>
7
- License: Apache License 2.0
8
- Project-URL: Repository, https://gitlab.com/mia_jinns/jinns
9
- Project-URL: Documentation, https://mia_jinns.gitlab.io/jinns/index.html
10
- Classifier: License :: OSI Approved :: Apache Software License
11
- Classifier: Development Status :: 4 - Beta
12
- Classifier: Programming Language :: Python
13
- Requires-Python: >=3.10
14
- Description-Content-Type: text/markdown
15
- License-File: LICENSE
16
- License-File: AUTHORS
17
- Requires-Dist: numpy
18
- Requires-Dist: jax
19
- Requires-Dist: jaxopt
20
- Requires-Dist: optax
21
- Requires-Dist: equinox >0.11.3
22
- Requires-Dist: jax-tqdm
23
- Requires-Dist: diffrax
24
- Requires-Dist: matplotlib
25
- Provides-Extra: notebook
26
- Requires-Dist: jupyter ; extra == 'notebook'
27
- Requires-Dist: seaborn ; extra == 'notebook'
28
-
29
- jinns
30
- =====
31
-
32
- Physics Informed Neural Networks with JAX. **jinns** has been developed to estimate solutions to your ODE et PDE problems using neural networks.
33
- **jinns** is built on JAX.
34
-
35
- **jinns** specific points:
36
-
37
- - **jinns** is coded with JAX as a backend: forward and backward autodiff, vmapping, jitting and more!
38
-
39
- - In **jinns**, we give the user maximum control on what is happening. We also keep the maths and computations visible and not hidden behind layers of code!
40
-
41
- - In the near future, we want to focus the development on inverse problems and inference in mecanistic-statistical models
42
-
43
- - [Separable PINNs](https://openreview.net/pdf?id=dEySGIcDnI) are implemented
44
-
45
- - [Hyper PINNs](https://arxiv.org/pdf/2111.01008.pdf) are implemented
46
-
47
- - Check out our various notebooks to get started with `jinns`
48
-
49
- For more information, open an issue or contact us!
50
-
51
- # Installation
52
-
53
- Install the latest version with pip
54
-
55
- ```bash
56
- pip install jinns
57
- ```
58
-
59
- # Documentation
60
-
61
- The project's documentation is available at [https://mia_jinns.gitlab.io/jinns/index.html](https://mia_jinns.gitlab.io/jinns/index.html)
62
-
63
- # Contributing
64
-
65
- * First fork the library on Gitlab.
66
-
67
- * Then clone and install the library in development mode with
68
-
69
- ```bash
70
- pip install -e .
71
- ```
72
-
73
- * Install pre-commit and run it.
74
-
75
- ```bash
76
- pip install pre-commit
77
- pre-commit install
78
- ```
79
-
80
- * Open a merge request once you are done with your changes.
81
-
82
- # Contributors & references
83
-
84
- *Active*: Hugo Gangloff, Nicolas Jouvin
85
- *Past*: Pierre Gloaguen, Charles Ollion, Achille Thin
jinns-1.1.0.dist-info/RECORD DELETED
@@ -1,39 +0,0 @@
1
- jinns/__init__.py,sha256=5p7V5VJd7PXEINqhqS4mUsnQtXlyPwfctRhL4p0loFg,181
2
- jinns/data/_Batchs.py,sha256=BLxTDiFb6o9M6Irc2_HKmpr8IgA159u_kJIbCBZ490E,926
3
- jinns/data/_DataGenerators.py,sha256=TuQDPI8NGx4WnCGtUp8v7o7GcnDfVZvL5E34JI-9Lmw,58455
4
- jinns/data/__init__.py,sha256=TRCH0Z4-SQZ50MbSf46CUYWBkWVDmXCyez9T-EGiv_8,338
5
- jinns/experimental/__init__.py,sha256=3jCIy2R2i_0Erwxg-HwISdH79Nt1XCXhS9yY1F5awiY,208
6
- jinns/experimental/_diffrax_solver.py,sha256=upMr3kTTNrxEiSUO_oLvCXcjS9lPxSjvbB81h3qlhaU,6813
7
- jinns/loss/_DynamicLoss.py,sha256=WGbAuWnNfsbzUlWEiW_ARd4kI3jmHwdqPjxLC-wCA6s,25753
8
- jinns/loss/_DynamicLossAbstract.py,sha256=Xyt28Oej_zlhcV3f6cw2vnAKyRJhXBiA63CsdL3PihU,13767
9
- jinns/loss/_LossODE.py,sha256=ThWPse6Gn5crM3_tzwZCBx-usoD0xWu6y1n0GVl2dpI,23422
10
- jinns/loss/_LossPDE.py,sha256=R9kNQiaFbFx2eCMdjB7ie7UJ9pJW7PmvHijioNgu-bs,49117
11
- jinns/loss/__init__.py,sha256=Fm4QAHaVmp0CA7HSwb7KUctwdXnNZ9v5KmTqpeoYPaE,669
12
- jinns/loss/_boundary_conditions.py,sha256=O0D8eWsFfvNNeO20PQ0rUKBI_MDqaBvqChfXaztZoL4,16679
13
- jinns/loss/_loss_utils.py,sha256=44J-VF6dxT_o5BcNWFOiLpY40c35YnAxxZkoNtdtcZc,13689
14
- jinns/loss/_loss_weights.py,sha256=F0Fgji2XpVk3pr9oIryGuXcG1FGQo4Dv6WFgze2BtA0,2201
15
- jinns/loss/_operators.py,sha256=o-Ljp_9_HXB9Mhm-ANh6ouNw4_PsqLJAha7dFDGl_nQ,10781
16
- jinns/parameters/__init__.py,sha256=1gxNLoAXUjhUzBWuh86YjU5pYy8SOboCs8TrKcU1wZc,158
17
- jinns/parameters/_derivative_keys.py,sha256=UyEcgfNF1vwPcGWD2ShAZkZiq4thzRDm_OUJzOfjjiY,21909
18
- jinns/parameters/_params.py,sha256=wK9ZSqoL9KnjOWqc_ZhJ09ffbsgeUEcttc1Rhme0lLk,3550
19
- jinns/plot/__init__.py,sha256=Q279h5veYWNLQyttsC8_tDOToqUHh8WaRON90CiWXqk,81
20
- jinns/plot/_plot.py,sha256=ZGIJdGwEd3NlHRTq_2sOfEH_CtOkvPwdgCMct-nQlJE,11691
21
- jinns/solver/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
22
- jinns/solver/_rar.py,sha256=BAr0qHNtCQ11XRhxRDed8JZPSWBm4E5FT3VOGQPrm4g,22728
23
- jinns/solver/_solve.py,sha256=iUkvZ9f-0bxO2eJlNv7wLk5oES69LNbDxzOgBAjbHTg,21106
24
- jinns/utils/__init__.py,sha256=CNxcb_AYzA2aeDsYwLfIuT1zy8NU7LElU160eHCj2oA,174
25
- jinns/utils/_containers.py,sha256=_PLIkJHY-jE3nigjuAiYE5USJr7rpXCDsNULRKZtqnU,1213
26
- jinns/utils/_hyperpinn.py,sha256=7fAy6Xv7CkIeIYCw7u9RB0gh4aITMfxz6JCmJ685KB8,16351
27
- jinns/utils/_pinn.py,sha256=eiw_i72D7-CxZXXOBpgOgDndwGd5sSnFZNk-Rg-6xy8,12977
28
- jinns/utils/_save_load.py,sha256=kVDeQrpPf7j7kYUy1gWGCr4_QyaBplRbPSitkWTQnQA,8574
29
- jinns/utils/_spinn.py,sha256=lesBjXwoj3UUzfecPdh1wBFCY0BlA-q0Wb1pDv0RVYA,9211
30
- jinns/utils/_types.py,sha256=P_dS0odrHbyalYJ0FjS6q0tkXAGr-4GArsiyJYrB1ho,1878
31
- jinns/utils/_utils.py,sha256=Ow8xB516E7yHDZatokVJHHFNPDu6fXr9-NmraUXjjyw,1819
32
- jinns/validation/__init__.py,sha256=Jv58mzgC3F7cRfXA6caicL1t_U0UAhbwLrmMNVg6E7s,66
33
- jinns/validation/_validation.py,sha256=bvqL2poTFJfn9lspWqMqXvQGcQIodKwKrC786QtEZ7A,4700
34
- jinns-1.1.0.dist-info/AUTHORS,sha256=7NwCj9nU-HNG1asvy4qhQ2w7oZHrn-Lk5_wK_Ve7a3M,80
35
- jinns-1.1.0.dist-info/LICENSE,sha256=BIAkGtXB59Q_BG8f6_OqtQ1BHPv60ggE9mpXJYz2dRM,11337
36
- jinns-1.1.0.dist-info/METADATA,sha256=3Qk885oguf6S_WPHd9KCFVIWP21nJqX9zFWoS9ZI-T0,2536
37
- jinns-1.1.0.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
38
- jinns-1.1.0.dist-info/top_level.txt,sha256=RXbkr2hzy8WBE8aiRyrJYFqn3JeMJIhMdybLjjLTB9c,6
39
- jinns-1.1.0.dist-info/RECORD,,
{jinns-1.1.0.dist-info → jinns-1.2.0.dist-info}/AUTHORS RENAMED
File without changes
{jinns-1.1.0.dist-info → jinns-1.2.0.dist-info}/LICENSE RENAMED
File without changes