jinns 1.0.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 (32) 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 +27 -73
  5. jinns/loss/_LossODE.py +45 -26
  6. jinns/loss/_LossPDE.py +85 -84
  7. jinns/loss/__init__.py +7 -6
  8. jinns/loss/_boundary_conditions.py +148 -279
  9. jinns/loss/_loss_utils.py +85 -58
  10. jinns/loss/_operators.py +441 -184
  11. jinns/parameters/_derivative_keys.py +487 -60
  12. jinns/plot/_plot.py +111 -98
  13. jinns/solver/_rar.py +102 -407
  14. jinns/solver/_solve.py +73 -38
  15. jinns/solver/_utils.py +122 -0
  16. jinns/utils/__init__.py +2 -0
  17. jinns/utils/_containers.py +3 -1
  18. jinns/utils/_hyperpinn.py +17 -7
  19. jinns/utils/_pinn.py +17 -27
  20. jinns/utils/_ppinn.py +227 -0
  21. jinns/utils/_save_load.py +13 -13
  22. jinns/utils/_spinn.py +24 -43
  23. jinns/utils/_types.py +1 -0
  24. jinns/utils/_utils.py +40 -12
  25. jinns-1.2.0.dist-info/AUTHORS +2 -0
  26. jinns-1.2.0.dist-info/METADATA +127 -0
  27. jinns-1.2.0.dist-info/RECORD +41 -0
  28. {jinns-1.0.0.dist-info → jinns-1.2.0.dist-info}/WHEEL +1 -1
  29. jinns-1.0.0.dist-info/METADATA +0 -84
  30. jinns-1.0.0.dist-info/RECORD +0 -38
  31. {jinns-1.0.0.dist-info → jinns-1.2.0.dist-info}/LICENSE +0 -0
  32. {jinns-1.0.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.0.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.0.0.dist-info/METADATA DELETED
@@ -1,84 +0,0 @@
1
- Metadata-Version: 2.1
2
- Name: jinns
3
- Version: 1.0.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
- Requires-Dist: numpy
17
- Requires-Dist: jax
18
- Requires-Dist: jaxopt
19
- Requires-Dist: optax
20
- Requires-Dist: equinox >0.11.3
21
- Requires-Dist: jax-tqdm
22
- Requires-Dist: diffrax
23
- Requires-Dist: matplotlib
24
- Provides-Extra: notebook
25
- Requires-Dist: jupyter ; extra == 'notebook'
26
- Requires-Dist: seaborn ; extra == 'notebook'
27
-
28
- jinns
29
- =====
30
-
31
- Physics Informed Neural Networks with JAX. **jinns** has been developed to estimate solutions to your ODE et PDE problems using neural networks.
32
- **jinns** is built on JAX.
33
-
34
- **jinns** specific points:
35
-
36
- - **jinns** is coded with JAX as a backend: forward and backward autodiff, vmapping, jitting and more!
37
-
38
- - 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!
39
-
40
- - In the near future, we want to focus the development on inverse problems and inference in mecanistic-statistical models
41
-
42
- - [Separable PINNs](https://openreview.net/pdf?id=dEySGIcDnI) are implemented
43
-
44
- - [Hyper PINNs](https://arxiv.org/pdf/2111.01008.pdf) are implemented
45
-
46
- - Check out our various notebooks to get started with `jinns`
47
-
48
- For more information, open an issue or contact us!
49
-
50
- # Installation
51
-
52
- Install the latest version with pip
53
-
54
- ```bash
55
- pip install jinns
56
- ```
57
-
58
- # Documentation
59
-
60
- The project's documentation is available at [https://mia_jinns.gitlab.io/jinns/index.html](https://mia_jinns.gitlab.io/jinns/index.html)
61
-
62
- # Contributing
63
-
64
- * First fork the library on Gitlab.
65
-
66
- * Then clone and install the library in development mode with
67
-
68
- ```bash
69
- pip install -e .
70
- ```
71
-
72
- * Install pre-commit and run it.
73
-
74
- ```bash
75
- pip install pre-commit
76
- pre-commit install
77
- ```
78
-
79
- * Open a merge request once you are done with your changes.
80
-
81
- # Contributors & references
82
-
83
- *Active*: Hugo Gangloff, Nicolas Jouvin
84
- *Past*: Pierre Gloaguen, Charles Ollion, Achille Thin
jinns-1.0.0.dist-info/RECORD DELETED
@@ -1,38 +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=UGFAZWCxbk7yTLdngNasaVawz51xZhu7LfCg4MTqj74,13765
9
- jinns/loss/_LossODE.py,sha256=4Tcl_Fpn7htoJvggreL3vv5xfQw2VR1tshLGVX7BUx4,22480
10
- jinns/loss/_LossPDE.py,sha256=zmnLblA6tK-KUKHFbwoo_N01CU1KwCdZt5lUCo-_ecM,48528
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=DI9KifH1aObrFau86BgAjtBNS8ecSRYB-_H6fduzGNg,13339
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=MbbyywuemjvgX-QVG2SSqlEreB7bMDCLnXmRZuGFkjE,3294
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.0.0.dist-info/LICENSE,sha256=BIAkGtXB59Q_BG8f6_OqtQ1BHPv60ggE9mpXJYz2dRM,11337
35
- jinns-1.0.0.dist-info/METADATA,sha256=XD9Az0b4tUbBNS1xReORk0j1HMkonZGs9_B0FfHFq34,2514
36
- jinns-1.0.0.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
37
- jinns-1.0.0.dist-info/top_level.txt,sha256=RXbkr2hzy8WBE8aiRyrJYFqn3JeMJIhMdybLjjLTB9c,6
38
- jinns-1.0.0.dist-info/RECORD,,
{jinns-1.0.0.dist-info → jinns-1.2.0.dist-info}/LICENSE RENAMED
File without changes