jinns 1.6.1__py3-none-any.whl → 1.7.1__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.
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: jinns
3
- Version: 1.6.1
3
+ Version: 1.7.1
4
4
  Summary: Physics Informed Neural Network with JAX
5
5
  Author-email: Hugo Gangloff <hugo.gangloff@inrae.fr>, Nicolas Jouvin <nicolas.jouvin@inrae.fr>
6
6
  Maintainer-email: Hugo Gangloff <hugo.gangloff@inrae.fr>, Nicolas Jouvin <nicolas.jouvin@inrae.fr>
@@ -8,23 +8,36 @@ License: Apache License 2.0
8
8
  Project-URL: Repository, https://gitlab.com/mia_jinns/jinns
9
9
  Project-URL: Documentation, https://mia_jinns.gitlab.io/jinns/index.html
10
10
  Classifier: License :: OSI Approved :: Apache Software License
11
- Classifier: Development Status :: 4 - Beta
11
+ Classifier: Development Status :: 5 - Production/Stable
12
12
  Classifier: Programming Language :: Python
13
13
  Requires-Python: >=3.11
14
14
  Description-Content-Type: text/markdown
15
15
  License-File: LICENSE
16
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
17
+ Requires-Dist: numpy>=2.0.0
18
+ Requires-Dist: jax>=0.8.1
19
+ Requires-Dist: optax>=0.2.6
20
+ Requires-Dist: equinox>=0.13.2
24
21
  Requires-Dist: matplotlib
22
+ Requires-Dist: jaxtyping
23
+ Requires-Dist: pyright>=1.1.407
24
+ Requires-Dist: pytest>=9.0.2
25
25
  Provides-Extra: notebook
26
26
  Requires-Dist: jupyter; extra == "notebook"
27
27
  Requires-Dist: seaborn; extra == "notebook"
28
+ Requires-Dist: pandas; extra == "notebook"
29
+ Requires-Dist: diffrax; extra == "notebook"
30
+ Provides-Extra: gpu-cuda13
31
+ Requires-Dist: jax[cuda13]>=0.8.1; extra == "gpu-cuda13"
32
+ Provides-Extra: gpu-cuda12
33
+ Requires-Dist: jax[cuda12]>=0.8.1; extra == "gpu-cuda12"
34
+ Provides-Extra: cpu
35
+ Requires-Dist: jax>=0.8.1; extra == "cpu"
36
+ Provides-Extra: dev
37
+ Requires-Dist: pre-commit; extra == "dev"
38
+ Requires-Dist: pytest; extra == "dev"
39
+ Requires-Dist: pylint; extra == "dev"
40
+ Requires-Dist: ruff; extra == "dev"
28
41
  Dynamic: license-file
29
42
 
30
43
  jinns
@@ -32,12 +45,11 @@ jinns
32
45
 
33
46
  ![status](https://gitlab.com/mia_jinns/jinns/badges/main/pipeline.svg) ![coverage](https://gitlab.com/mia_jinns/jinns/badges/main/coverage.svg)
34
47
 
35
- Physics Informed Neural Networks with JAX. **jinns** is developed to estimate solutions of ODE and PDE problems using neural networks, with a strong focus on
48
+ Physics Informed Neural Networks with JAX. **jinns** is a Python package for physics-informed neural networks (PINNs) in the [JAX](https://jax.readthedocs.io/en/latest/) ecosystem. It provides an intuitive and flexible interface for
36
49
 
37
- 1. inverse problems: find equation parameters given noisy/indirect observations
38
- 2. meta-modeling: solve for a parametric family of differential equations
39
-
40
- It can also be used for forward problems and hybrid-modeling.
50
+ * forward problem: learning a PDE solution.
51
+ * inverse problem: learning the parameters of a PDE. **New in jinns v1.7.0:** `jinns.solve_alternate()` for fine-grained and efficient inverse problems.
52
+ * meta-modeling: learning a family of PDE indexed by its parameters.
41
53
 
42
54
  **jinns** specific points:
43
55
 
@@ -0,0 +1,58 @@
1
+ jinns/__init__.py,sha256=f8ZCZH95C7U8NfJ3M-00kf1IBbdgbwRNBhlluiOCGJE,530
2
+ jinns/data/_AbstractDataGenerator.py,sha256=le5plNhOE7hV72SC6p2xhWxljnGeStPc2kitzeMR_ts,535
3
+ jinns/data/_Batchs.py,sha256=mtHDlsen1vBBmFgGzynPmFfmBFC7zKcnnVEg_JnNlig,2857
4
+ jinns/data/_CubicMeshPDENonStatio.py,sha256=dtev6j7HuVZ_IBGnKVKOlJTyZVjnW_8G1VRGpcL40VU,22922
5
+ jinns/data/_CubicMeshPDEStatio.py,sha256=g4H9wI0FkA9sdnlSxp00QArq9IjhPrmsGgSleLdNpI4,23314
6
+ jinns/data/_DataGeneratorODE.py,sha256=7m3mhMk-o02Jq_ZieWktFG5w6mVwnWPs1XYiYqoJ0DA,7731
7
+ jinns/data/_DataGeneratorObservations.py,sha256=Lw1Fno4UDJ0LRkhFJ0tDtkkW-yWJLeswCWja504pUDo,27781
8
+ jinns/data/_DataGeneratorParameter.py,sha256=m7WCWBtmNptF8qQSWGPTQEFVzkJb8kJpidKAFBmoBiQ,10205
9
+ jinns/data/__init__.py,sha256=DEzEmoD5TmjHCaG6pS2jmFDzuhZn1ZDpFdVa2v0jCe0,703
10
+ jinns/data/_utils.py,sha256=xKHf7NksJ_AmrtEcpJsh7WSEvI3Yk98_cM5kmXSfmx0,5596
11
+ jinns/experimental/__init__.py,sha256=DT9e57zbjfzPeRnXemGUqnGd--MhV77FspChT0z4YrE,410
12
+ jinns/experimental/_diffrax_solver.py,sha256=upMr3kTTNrxEiSUO_oLvCXcjS9lPxSjvbB81h3qlhaU,6813
13
+ jinns/loss/_DynamicLoss.py,sha256=FXVraH97tIWA5ByO-V-r54Nhwd_EvHOfr0e2qjrfdf0,22667
14
+ jinns/loss/_DynamicLossAbstract.py,sha256=mm687ly-r7ONZPJaQN6kQa9CwM05msiZt9cCkQ4Ue54,13699
15
+ jinns/loss/_LossODE.py,sha256=mbsK6n61J3Fx4zl1VvJCLigowV-UPxIJoxn-Zf_IwH0,18039
16
+ jinns/loss/_LossPDE.py,sha256=rjy5mKEkFOrTqpRf7W51JEVnIysnD7RhQ33aAvZMcfg,43056
17
+ jinns/loss/__init__.py,sha256=z5xYgBipNFf66__5BqQc6R_8r4F6A3TXL60YjsM8Osk,1287
18
+ jinns/loss/_abstract_loss.py,sha256=MG-nC_ApYNhnTGLvW1_vx8VXXJLJ2q3Ird9sbUZN5fk,9000
19
+ jinns/loss/_boundary_conditions.py,sha256=WqPAIcG5DbY-K_sYYvi_cr4TPoi0ajIGvnmz9KSw3IE,12458
20
+ jinns/loss/_loss_components.py,sha256=Y5GGNuAS_tHl-AzIMBxf3yKj0N2QCdVdTY4jC7NDuic,442
21
+ jinns/loss/_loss_utils.py,sha256=thnV4rA77tH-exwe5Y2dbPOTDAmw44sQJmYyRGe8cRY,12007
22
+ jinns/loss/_loss_weight_updates.py,sha256=WaQZl4TE1eXybLGjMXaXRQhUtA_7Cnyc1r5vTyr4_98,7683
23
+ jinns/loss/_loss_weights.py,sha256=mPfIXdCoTAcnxsgNK_KjMTBUbk_Q3m4BmFF229ghEpQ,2640
24
+ jinns/loss/_operators.py,sha256=RsBJzvvADB1uhJQSbvg1e90KPw8csQR5GKIidu5Xsp8,22874
25
+ jinns/nn/__init__.py,sha256=gwE48oqB_FsSIE-hUvCLz0jPaqX350LBxzH6ueFWYk4,456
26
+ jinns/nn/_abstract_pinn.py,sha256=jp1Wi2LDBvSkJempliQq4m7OaIbYr8d16elnJX8movk,600
27
+ jinns/nn/_hyperpinn.py,sha256=A4nNnID7XVsU7kDafIZo7OchRx9EmoSHseoH4EmY0m8,20264
28
+ jinns/nn/_mlp.py,sha256=VqObd7gLGIO25N4XVfHK_XwvhR2dgeTR-kOcRLTH0Os,8931
29
+ jinns/nn/_pinn.py,sha256=f5OyiZCgyq_wubI10rNYSlY00letMsXYDSHcISuvvwM,8322
30
+ jinns/nn/_ppinn.py,sha256=wvzeT8CyYeiQ2q6BTvNTz3L7g7FPemOQcPIGVDy34Rs,10136
31
+ jinns/nn/_save_load.py,sha256=4pGDPZPKzDmY5UzRtA_zxpDK4hFL2LuDxexWW2PnHtw,7656
32
+ jinns/nn/_spinn.py,sha256=MGaPQRA-qVltshESB7IP_kft0kINjijPhEV5kBfKDvI,4346
33
+ jinns/nn/_spinn_mlp.py,sha256=NWSNqVOUPz-Zk2JNYZKbmcYWtXHDrxN2aIYiJH4jK1k,7641
34
+ jinns/nn/_utils.py,sha256=SoaYvNM6V7M3GcXK3diqO7B9_zoBXqSux2fsTOcouSA,1018
35
+ jinns/parameters/__init__.py,sha256=T_BhKT77X5nei2pYj-QIOKCyD6j2UCgSgAY3LnNdkCo,333
36
+ jinns/parameters/_derivative_keys.py,sha256=laMZoT7ZZa3ZSTuvA-TKM6-VX32vKZFfA5ThDWhj4v4,21084
37
+ jinns/parameters/_params.py,sha256=A6F5HAbxaKPonsH5IrT3ERNSy60OGLUD0eV9hYNj28U,4245
38
+ jinns/plot/__init__.py,sha256=KPHX0Um4FbciZO1yD8kjZbkaT8tT964Y6SE2xCQ4eDU,135
39
+ jinns/plot/_plot.py,sha256=-A5auNeElaz2_8UzVQJQE4143ZFg0zgMjStU7kwttEY,11565
40
+ jinns/solver/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
41
+ jinns/solver/_rar.py,sha256=IFaH0-b-uPp7fs1rX14LY2oy3cXH2NN6QqjC4rdm6n8,10659
42
+ jinns/solver/_solve.py,sha256=3JDlX19o3UCpIfrTyFLgWh3ihAwGbl1oB0NsdbBke30,20130
43
+ jinns/solver/_solve_alternate.py,sha256=9R1idmQMhkLz_QsxwjsfWcS_f_Cvg71NS43MfNbpXSo,31664
44
+ jinns/solver/_utils.py,sha256=UMq7jhrShVeVUbcJqvjNdLvczDOP1x6XolF0jWjNs0A,23002
45
+ jinns/utils/_DictToModuleMeta.py,sha256=kdCAZ6U7L_ZNsGnvTfk3r-85RaV19Qq6y6zdhvI_hGM,3043
46
+ jinns/utils/_ItemizableModule.py,sha256=KdjsisFddGOUTA7-1jXlrkFxnX5sIVdtyNDxOaEcEYw,634
47
+ jinns/utils/__init__.py,sha256=k2kWDahXsLhSlsbeB_W73WLaVSvPG_S6mrNG_5fW_Mo,121
48
+ jinns/utils/_containers.py,sha256=t6okFDd0Sq6R3ev1tdc-_cMKtGwAx8G4-TUNBb1ojtc,2009
49
+ jinns/utils/_types.py,sha256=PftmBr-hV33U-aB_TzTYNLD42ogRWmZKA02CPdxQWBg,2482
50
+ jinns/utils/_utils.py,sha256=M7NXX9ok-BkH5o_xo74PB1_Cc8XiDipSl51rq82dTH4,2821
51
+ jinns/validation/__init__.py,sha256=FTyUO-v1b8Tv-FDSQsntrH7zl9E0ENexqKMT_dFRkYo,124
52
+ jinns/validation/_validation.py,sha256=8p6sMKiBAvA6JNm65hjkMj0997LJ0BkyCREEh0AnPVE,4803
53
+ jinns-1.7.1.dist-info/licenses/AUTHORS,sha256=7NwCj9nU-HNG1asvy4qhQ2w7oZHrn-Lk5_wK_Ve7a3M,80
54
+ jinns-1.7.1.dist-info/licenses/LICENSE,sha256=BIAkGtXB59Q_BG8f6_OqtQ1BHPv60ggE9mpXJYz2dRM,11337
55
+ jinns-1.7.1.dist-info/METADATA,sha256=BuWK-l7JNmKFJ9silTxgpeJQqNgX3JRLZfo0WgmGb4Y,5982
56
+ jinns-1.7.1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
57
+ jinns-1.7.1.dist-info/top_level.txt,sha256=RXbkr2hzy8WBE8aiRyrJYFqn3JeMJIhMdybLjjLTB9c,6
58
+ jinns-1.7.1.dist-info/RECORD,,
@@ -1,5 +1,5 @@
1
1
  Wheel-Version: 1.0
2
- Generator: setuptools (80.9.0)
2
+ Generator: setuptools (80.10.2)
3
3
  Root-Is-Purelib: true
4
4
  Tag: py3-none-any
5
5
 
@@ -1,57 +0,0 @@
1
- jinns/__init__.py,sha256=Tjp5z0Mnd1nscvJaXnxZb4lEYbI0cpN6OPgCZ-Swo74,453
2
- jinns/data/_AbstractDataGenerator.py,sha256=le5plNhOE7hV72SC6p2xhWxljnGeStPc2kitzeMR_ts,535
3
- jinns/data/_Batchs.py,sha256=UZfJWIKAFLQJxOtWN1ybJdoSQCO6PWk3SAgV9YmNVnI,2809
4
- jinns/data/_CubicMeshPDENonStatio.py,sha256=dtev6j7HuVZ_IBGnKVKOlJTyZVjnW_8G1VRGpcL40VU,22922
5
- jinns/data/_CubicMeshPDEStatio.py,sha256=g4H9wI0FkA9sdnlSxp00QArq9IjhPrmsGgSleLdNpI4,23314
6
- jinns/data/_DataGeneratorODE.py,sha256=XOGeX9m5J9CkkGx2alJ5G7FZYGn5cBUaFpO-3-MKvdc,7724
7
- jinns/data/_DataGeneratorObservations.py,sha256=h2bS-oOUi1rYvhDHy8xT-N9voqtb_IOelL2asnXCybw,9765
8
- jinns/data/_DataGeneratorParameter.py,sha256=m7WCWBtmNptF8qQSWGPTQEFVzkJb8kJpidKAFBmoBiQ,10205
9
- jinns/data/__init__.py,sha256=DEzEmoD5TmjHCaG6pS2jmFDzuhZn1ZDpFdVa2v0jCe0,703
10
- jinns/data/_utils.py,sha256=xKHf7NksJ_AmrtEcpJsh7WSEvI3Yk98_cM5kmXSfmx0,5596
11
- jinns/experimental/__init__.py,sha256=DT9e57zbjfzPeRnXemGUqnGd--MhV77FspChT0z4YrE,410
12
- jinns/experimental/_diffrax_solver.py,sha256=upMr3kTTNrxEiSUO_oLvCXcjS9lPxSjvbB81h3qlhaU,6813
13
- jinns/loss/_DynamicLoss.py,sha256=FXVraH97tIWA5ByO-V-r54Nhwd_EvHOfr0e2qjrfdf0,22667
14
- jinns/loss/_DynamicLossAbstract.py,sha256=bTiDZHqkHDdvfj6hcjPcESYzwzqOd2LtR_h7GPAqQpY,13602
15
- jinns/loss/_LossODE.py,sha256=iIW9IGlmPH8coAGe28ysDC75PGrjrRM4YnJN16Cqq3o,15029
16
- jinns/loss/_LossPDE.py,sha256=nScI0gOB6mK1r5oq-KfAm_Wmcb_frOFc1eyjLnCaYqA,38491
17
- jinns/loss/__init__.py,sha256=z5xYgBipNFf66__5BqQc6R_8r4F6A3TXL60YjsM8Osk,1287
18
- jinns/loss/_abstract_loss.py,sha256=yUaulxRL9Pc6P2J6icP_-In-yoeX-yRsOla8kzixxjE,6285
19
- jinns/loss/_boundary_conditions.py,sha256=9HGw1cGLfmEilP4V4B2T0zl0YP1kNtrtXVLQNiBmWgc,12464
20
- jinns/loss/_loss_components.py,sha256=Y5GGNuAS_tHl-AzIMBxf3yKj0N2QCdVdTY4jC7NDuic,442
21
- jinns/loss/_loss_utils.py,sha256=eJ4JcBm396LHx7Tti88ZQrLcKqVL1oSfFGT23VNkytQ,11949
22
- jinns/loss/_loss_weight_updates.py,sha256=Q5RJkulMjP5G3twKEGmc46QeQJSCb7yyS7OudzRqQkY,6775
23
- jinns/loss/_loss_weights.py,sha256=x0URuCaFXULlVDwlL9ZEy45L33tj1Fa0VJUQlSPhGkA,2416
24
- jinns/loss/_operators.py,sha256=Ds5yRH7hu-jaGRp7PYbt821BgYuEvgWHufWhYgdMjw0,22909
25
- jinns/nn/__init__.py,sha256=gwE48oqB_FsSIE-hUvCLz0jPaqX350LBxzH6ueFWYk4,456
26
- jinns/nn/_abstract_pinn.py,sha256=E17TFjxgs63BHl3dUKwajqkBuggVbyNVtovX0ePPzr4,602
27
- jinns/nn/_hyperpinn.py,sha256=x_hBWqokBbURMOdIG_0EBNGJjxAzjlUZ4__fyb-uasA,20245
28
- jinns/nn/_mlp.py,sha256=ajVWRTtGVnmRjuaylEMlsK-3H5LMqKDvhUsMNnH_rIM,8935
29
- jinns/nn/_pinn.py,sha256=nRBIdjTir1ViI9hBO4_4DY6uS-1P8Xv-TDLU-O_qzTk,8332
30
- jinns/nn/_ppinn.py,sha256=qBMenzj0TpapGxyl7nSWb6mER-Ny42JrcOd7Wld19hU,10144
31
- jinns/nn/_save_load.py,sha256=4pGDPZPKzDmY5UzRtA_zxpDK4hFL2LuDxexWW2PnHtw,7656
32
- jinns/nn/_spinn.py,sha256=a0ZsnzyXi-oNd6i8pAwF1tdjJQU51pbrGg608O6ykQw,4352
33
- jinns/nn/_spinn_mlp.py,sha256=7w-WsdcoLHVWwkK2DtXRKRtSrcJkEXJ2_n27VmXI2j4,7651
34
- jinns/nn/_utils.py,sha256=SoaYvNM6V7M3GcXK3diqO7B9_zoBXqSux2fsTOcouSA,1018
35
- jinns/parameters/__init__.py,sha256=T_BhKT77X5nei2pYj-QIOKCyD6j2UCgSgAY3LnNdkCo,333
36
- jinns/parameters/_derivative_keys.py,sha256=snyUGSX7RRrQ6kwtxZPS5U4stWj5Nj2RQ4RABGcInwo,20652
37
- jinns/parameters/_params.py,sha256=kU_rql_Wb7MJQsBDDcIwOf7RZrxkMRikP1w5xpvnmVs,3953
38
- jinns/plot/__init__.py,sha256=KPHX0Um4FbciZO1yD8kjZbkaT8tT964Y6SE2xCQ4eDU,135
39
- jinns/plot/_plot.py,sha256=-A5auNeElaz2_8UzVQJQE4143ZFg0zgMjStU7kwttEY,11565
40
- jinns/solver/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
41
- jinns/solver/_rar.py,sha256=vSVTnCGCusI1vTZCvIkP2_G8we44G_42yZHx2sOK9DE,10291
42
- jinns/solver/_solve.py,sha256=XG7G0OdVGExRCZ84JEzT74NefZRdUP4VFQoKkw9e-C4,28650
43
- jinns/solver/_utils.py,sha256=sM2UbVzYyjw24l4QSIR3IlynJTPGD_S08r8v0lXMxA8,5876
44
- jinns/utils/_DictToModuleMeta.py,sha256=V3D3rN6rCi6NjI8Cht0PTgeSgLe9ji3PsAX5y1-vbHY,2977
45
- jinns/utils/_ItemizableModule.py,sha256=KdjsisFddGOUTA7-1jXlrkFxnX5sIVdtyNDxOaEcEYw,634
46
- jinns/utils/__init__.py,sha256=k2kWDahXsLhSlsbeB_W73WLaVSvPG_S6mrNG_5fW_Mo,121
47
- jinns/utils/_containers.py,sha256=YShcrPKfj5_I9mn3NMAS4Ea9MhhyL7fjv0e3MRbITHg,1837
48
- jinns/utils/_types.py,sha256=mmDKnZavLBRNHrfsa4w2KgZbm-3Xrlu3Vx_mJ9sKI78,1288
49
- jinns/utils/_utils.py,sha256=M7NXX9ok-BkH5o_xo74PB1_Cc8XiDipSl51rq82dTH4,2821
50
- jinns/validation/__init__.py,sha256=FTyUO-v1b8Tv-FDSQsntrH7zl9E0ENexqKMT_dFRkYo,124
51
- jinns/validation/_validation.py,sha256=8p6sMKiBAvA6JNm65hjkMj0997LJ0BkyCREEh0AnPVE,4803
52
- jinns-1.6.1.dist-info/licenses/AUTHORS,sha256=7NwCj9nU-HNG1asvy4qhQ2w7oZHrn-Lk5_wK_Ve7a3M,80
53
- jinns-1.6.1.dist-info/licenses/LICENSE,sha256=BIAkGtXB59Q_BG8f6_OqtQ1BHPv60ggE9mpXJYz2dRM,11337
54
- jinns-1.6.1.dist-info/METADATA,sha256=f3RcBZX2apKaq1DNb243IOlI2ieoUr0l1a5BDw0Wfsc,5314
55
- jinns-1.6.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
56
- jinns-1.6.1.dist-info/top_level.txt,sha256=RXbkr2hzy8WBE8aiRyrJYFqn3JeMJIhMdybLjjLTB9c,6
57
- jinns-1.6.1.dist-info/RECORD,,