jinns 1.6.1__tar.gz → 1.7.1__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 (196) hide show
  1. {jinns-1.6.1 → jinns-1.7.1}/.gitignore +4 -0
  2. jinns-1.7.1/.pre-commit-config.yaml +25 -0
  3. jinns-1.7.1/Notebooks/ODE/1D_GLV_adaptative_loss_weights.ipynb +1251 -0
  4. {jinns-1.6.1 → jinns-1.7.1}/Notebooks/ODE/1D_Generalized_Lotka_Volterra.ipynb +6 -4
  5. jinns-1.7.1/Notebooks/ODE/GLV_JointEstimation_Alternate.ipynb +2332 -0
  6. jinns-1.7.1/Notebooks/ODE/linear_fo_equation.ipynb +624 -0
  7. {jinns-1.6.1 → jinns-1.7.1}/Notebooks/ODE/systems_biology_informed_neural_network.ipynb +22 -11
  8. jinns-1.7.1/Notebooks/ODE/utils_GLV_JointEstimation_Alternate.py +146 -0
  9. jinns-1.7.1/Notebooks/PDE/1D_non_stationary_Burgers.ipynb +807 -0
  10. {jinns-1.6.1 → jinns-1.7.1}/Notebooks/PDE/1D_non_stationary_Fisher_KPP_Bounded_Domain.ipynb +9 -12
  11. jinns-1.7.1/Notebooks/PDE/2D_Heat_inverse_problem.ipynb +932 -0
  12. {jinns-1.6.1 → jinns-1.7.1}/Notebooks/PDE/2D_Navier_Stokes_PipeFlow.ipynb +11 -9
  13. {jinns-1.6.1 → jinns-1.7.1}/Notebooks/PDE/2D_Navier_Stokes_PipeFlow_Metamodel_hyperpinn.ipynb +8 -5
  14. {jinns-1.6.1 → jinns-1.7.1}/Notebooks/PDE/2D_Navier_Stokes_PipeFlow_SoftConstraints.ipynb +11 -7
  15. {jinns-1.6.1 → jinns-1.7.1}/Notebooks/PDE/2D_Poisson_inverse_problem.ipynb +7 -5
  16. {jinns-1.6.1 → jinns-1.7.1}/Notebooks/PDE/2D_non_stationary_OU.ipynb +5 -5
  17. {jinns-1.6.1 → jinns-1.7.1}/Notebooks/PDE/Reaction_Diffusion_2D_heterogenous_model.ipynb +9 -9
  18. {jinns-1.6.1 → jinns-1.7.1}/Notebooks/PDE/Reaction_Diffusion_2D_homogeneous_metamodel_hyperpinn_diffrax.ipynb +4 -4
  19. {jinns-1.6.1 → jinns-1.7.1}/Notebooks/Tutorials/1D_non_stationary_Burgers_JointEstimation_Vanilla.ipynb +3 -4
  20. jinns-1.7.1/Notebooks/Tutorials/GLV_JointEstimation_Alternate.ipynb +1867 -0
  21. {jinns-1.6.1 → jinns-1.7.1}/Notebooks/Tutorials/implementing_your_own_PDE_problem.ipynb +24 -13
  22. {jinns-1.6.1 → jinns-1.7.1}/Notebooks/Tutorials/introducing_validation_loss.ipynb +8 -11
  23. {jinns-1.6.1 → jinns-1.7.1}/Notebooks/Tutorials/load_save_model.ipynb +4 -4
  24. jinns-1.7.1/Notebooks/Tutorials/tuto_GLV_inverse_problem.ipynb +1608 -0
  25. jinns-1.7.1/Notebooks/Tutorials/utils_GLV_JointEstimation_Alternate.py +148 -0
  26. {jinns-1.6.1/jinns.egg-info → jinns-1.7.1}/PKG-INFO +26 -14
  27. {jinns-1.6.1 → jinns-1.7.1}/README.md +4 -5
  28. jinns-1.7.1/docs/api/solver.md +15 -0
  29. {jinns-1.6.1 → jinns-1.7.1}/docs/changelog.md +8 -1
  30. {jinns-1.6.1 → jinns-1.7.1}/docs/index.md +2 -2
  31. jinns-1.7.1/img/jinns-diagram.png +0 -0
  32. {jinns-1.6.1 → jinns-1.7.1}/jinns/__init__.py +2 -1
  33. {jinns-1.6.1 → jinns-1.7.1}/jinns/data/_Batchs.py +4 -4
  34. {jinns-1.6.1 → jinns-1.7.1}/jinns/data/_DataGeneratorODE.py +1 -1
  35. jinns-1.7.1/jinns/data/_DataGeneratorObservations.py +651 -0
  36. {jinns-1.6.1 → jinns-1.7.1}/jinns/loss/_DynamicLossAbstract.py +3 -1
  37. {jinns-1.6.1 → jinns-1.7.1}/jinns/loss/_LossODE.py +138 -73
  38. {jinns-1.6.1 → jinns-1.7.1}/jinns/loss/_LossPDE.py +208 -104
  39. {jinns-1.6.1 → jinns-1.7.1}/jinns/loss/_abstract_loss.py +97 -14
  40. {jinns-1.6.1 → jinns-1.7.1}/jinns/loss/_boundary_conditions.py +6 -6
  41. {jinns-1.6.1 → jinns-1.7.1}/jinns/loss/_loss_utils.py +2 -2
  42. {jinns-1.6.1 → jinns-1.7.1}/jinns/loss/_loss_weight_updates.py +30 -0
  43. {jinns-1.6.1 → jinns-1.7.1}/jinns/loss/_loss_weights.py +4 -0
  44. {jinns-1.6.1 → jinns-1.7.1}/jinns/loss/_operators.py +27 -27
  45. {jinns-1.6.1 → jinns-1.7.1}/jinns/nn/_abstract_pinn.py +1 -1
  46. {jinns-1.6.1 → jinns-1.7.1}/jinns/nn/_hyperpinn.py +6 -6
  47. {jinns-1.6.1 → jinns-1.7.1}/jinns/nn/_mlp.py +3 -3
  48. {jinns-1.6.1 → jinns-1.7.1}/jinns/nn/_pinn.py +7 -7
  49. {jinns-1.6.1 → jinns-1.7.1}/jinns/nn/_ppinn.py +6 -6
  50. {jinns-1.6.1 → jinns-1.7.1}/jinns/nn/_spinn.py +4 -4
  51. {jinns-1.6.1 → jinns-1.7.1}/jinns/nn/_spinn_mlp.py +7 -7
  52. {jinns-1.6.1 → jinns-1.7.1}/jinns/parameters/_derivative_keys.py +13 -6
  53. {jinns-1.6.1 → jinns-1.7.1}/jinns/parameters/_params.py +10 -0
  54. {jinns-1.6.1 → jinns-1.7.1}/jinns/solver/_rar.py +19 -9
  55. {jinns-1.6.1 → jinns-1.7.1}/jinns/solver/_solve.py +102 -367
  56. jinns-1.7.1/jinns/solver/_solve_alternate.py +885 -0
  57. jinns-1.7.1/jinns/solver/_utils.py +629 -0
  58. {jinns-1.6.1 → jinns-1.7.1}/jinns/utils/_DictToModuleMeta.py +3 -1
  59. {jinns-1.6.1 → jinns-1.7.1}/jinns/utils/_containers.py +8 -4
  60. {jinns-1.6.1 → jinns-1.7.1}/jinns/utils/_types.py +42 -1
  61. {jinns-1.6.1 → jinns-1.7.1/jinns.egg-info}/PKG-INFO +26 -14
  62. {jinns-1.6.1 → jinns-1.7.1}/jinns.egg-info/SOURCES.txt +14 -1
  63. jinns-1.7.1/jinns.egg-info/requires.txt +29 -0
  64. {jinns-1.6.1 → jinns-1.7.1}/mkdocs.yml +4 -1
  65. {jinns-1.6.1 → jinns-1.7.1}/pyproject.toml +18 -5
  66. {jinns-1.6.1 → jinns-1.7.1}/tests/adaptative_weight_tests/test_ReLoBRaLo_update.py +5 -1
  67. {jinns-1.6.1 → jinns-1.7.1}/tests/adaptative_weight_tests/test_loss_weight_update.py +24 -25
  68. {jinns-1.6.1 → jinns-1.7.1}/tests/adaptative_weight_tests/test_lr_annealing.py +4 -0
  69. jinns-1.6.1/tests/adaptative_weight_tests/test_soft_adapt.py → jinns-1.7.1/tests/adaptative_weight_tests/test_prior_loss.py +6 -6
  70. jinns-1.7.1/tests/adaptative_weight_tests/test_soft_adapt.py +169 -0
  71. jinns-1.7.1/tests/adaptative_weight_tests/test_vectorial_lossODE_and_weight_update.py +205 -0
  72. jinns-1.7.1/tests/adaptative_weight_tests/test_vectorial_lossPDE_and_weight_update.py +260 -0
  73. jinns-1.7.1/tests/alternate_solver_tests/test_solve_alternate.py +165 -0
  74. jinns-1.7.1/tests/dataGenerator_tests/profiling_DataGeneratorObservations.py +110 -0
  75. jinns-1.7.1/tests/dataGenerator_tests/test_DataGeneratorObservations.py +303 -0
  76. {jinns-1.6.1 → jinns-1.7.1}/tests/loss_tests/test_lossODE.py +3 -3
  77. {jinns-1.6.1 → jinns-1.7.1}/tests/loss_tests/test_lossPDEnonstatio.py +1 -1
  78. {jinns-1.6.1 → jinns-1.7.1}/tests/loss_tests/test_lossPDEstatio.py +1 -1
  79. {jinns-1.6.1 → jinns-1.7.1}/tests/loss_tests/test_norm_loss.py +23 -23
  80. {jinns-1.6.1 → jinns-1.7.1}/tests/nn_tests/test_hyperpinns.py +4 -8
  81. {jinns-1.6.1 → jinns-1.7.1}/tests/nn_tests/test_mlp.py +3 -3
  82. {jinns-1.6.1 → jinns-1.7.1}/tests/nn_tests/test_pinn.py +4 -8
  83. {jinns-1.6.1 → jinns-1.7.1}/tests/nn_tests/test_ppinn_mlp.py +1 -1
  84. {jinns-1.6.1 → jinns-1.7.1}/tests/nn_tests/test_smlp.py +1 -1
  85. {jinns-1.6.1 → jinns-1.7.1}/tests/nn_tests/test_spinn.py +14 -21
  86. {jinns-1.6.1 → jinns-1.7.1}/tests/operator_tests/test_divergence_fwd.py +2 -2
  87. {jinns-1.6.1 → jinns-1.7.1}/tests/operator_tests/test_divergence_rev.py +2 -2
  88. {jinns-1.6.1 → jinns-1.7.1}/tests/operator_tests/test_laplacian_fwd.py +2 -2
  89. {jinns-1.6.1 → jinns-1.7.1}/tests/operator_tests/test_laplacian_rev.py +2 -2
  90. {jinns-1.6.1 → jinns-1.7.1}/tests/operator_tests/test_vectorial_laplacian_fwd.py +10 -10
  91. {jinns-1.6.1 → jinns-1.7.1}/tests/operator_tests/test_vectorial_laplacian_rev.py +10 -10
  92. {jinns-1.6.1 → jinns-1.7.1}/tests/parameters_tests/test_DerivativeKeysODE.py +45 -2
  93. {jinns-1.6.1 → jinns-1.7.1}/tests/plot_tests/test_plot1D.py +1 -1
  94. {jinns-1.6.1 → jinns-1.7.1}/tests/plot_tests/test_plot2D.py +1 -1
  95. {jinns-1.6.1 → jinns-1.7.1}/tests/save_load_tests/test_saving_loading_hyperpinn.py +1 -1
  96. {jinns-1.6.1 → jinns-1.7.1}/tests/save_load_tests/test_saving_loading_pinn.py +1 -1
  97. {jinns-1.6.1 → jinns-1.7.1}/tests/save_load_tests/test_saving_loading_spinn.py +2 -2
  98. jinns-1.7.1/tests/sharding_tests/test_sharding.py +163 -0
  99. {jinns-1.6.1 → jinns-1.7.1}/tests/solver_tests/test_Burgers_x32.py +2 -2
  100. {jinns-1.6.1 → jinns-1.7.1}/tests/solver_tests/test_Burgers_x64.py +2 -2
  101. {jinns-1.6.1 → jinns-1.7.1}/tests/solver_tests/test_Fisher_x32.py +2 -2
  102. {jinns-1.6.1 → jinns-1.7.1}/tests/solver_tests/test_Fisher_x64.py +2 -2
  103. {jinns-1.6.1 → jinns-1.7.1}/tests/solver_tests/test_GLV_x32.py +8 -6
  104. {jinns-1.6.1 → jinns-1.7.1}/tests/solver_tests/test_GLV_x64.py +1 -1
  105. {jinns-1.6.1 → jinns-1.7.1}/tests/solver_tests/test_NSPipeFlow_x32.py +2 -5
  106. {jinns-1.6.1 → jinns-1.7.1}/tests/solver_tests/test_NSPipeFlow_x64.py +2 -5
  107. {jinns-1.6.1 → jinns-1.7.1}/tests/solver_tests/test_OU1D_statio_x32.py +2 -2
  108. {jinns-1.6.1 → jinns-1.7.1}/tests/solver_tests/test_OU2D_x32.py +2 -2
  109. {jinns-1.6.1 → jinns-1.7.1}/tests/solver_tests/test_nan_params_catch.py +23 -14
  110. {jinns-1.6.1 → jinns-1.7.1}/tests/solver_tests/test_parameter_tracker.py +25 -12
  111. {jinns-1.6.1 → jinns-1.7.1}/tests/solver_tests/test_rar_algorithm.py +9 -7
  112. {jinns-1.6.1 → jinns-1.7.1}/tests/solver_tests_hyperpinn/test_NSPipeFlow_x32_hyperpinn.py +2 -2
  113. {jinns-1.6.1 → jinns-1.7.1}/tests/solver_tests_inverse/test_simple_inverse_pb.py +1 -1
  114. {jinns-1.6.1 → jinns-1.7.1}/tests/solver_tests_spinn/test_Burgers_x32_spinn.py +2 -2
  115. {jinns-1.6.1 → jinns-1.7.1}/tests/solver_tests_spinn/test_Fisher_x32_spinn.py +3 -3
  116. {jinns-1.6.1 → jinns-1.7.1}/tests/solver_tests_spinn/test_NSPipeFlow_x32_spinn.py +2 -5
  117. {jinns-1.6.1 → jinns-1.7.1}/tests/solver_tests_spinn/test_OU2D_x32_spinn.py +5 -5
  118. {jinns-1.6.1 → jinns-1.7.1}/tests/solver_tests_spinn/test_ReactionDiffusion_nonhomo_x64_spinn.py +3 -3
  119. {jinns-1.6.1 → jinns-1.7.1}/tests/validation_tests/test_vanilla_validation.py +7 -5
  120. jinns-1.6.1/.pre-commit-config.yaml +0 -11
  121. jinns-1.6.1/Notebooks/ODE/linear_fo_equation.ipynb +0 -539
  122. jinns-1.6.1/Notebooks/PDE/1D_non_stationary_Burgers.ipynb +0 -785
  123. jinns-1.6.1/Notebooks/PDE/2D_Heat_inverse_problem.ipynb +0 -931
  124. jinns-1.6.1/docs/api/solver.md +0 -7
  125. jinns-1.6.1/img/jinns-diagram.png +0 -0
  126. jinns-1.6.1/jinns/data/_DataGeneratorObservations.py +0 -243
  127. jinns-1.6.1/jinns/solver/_utils.py +0 -120
  128. jinns-1.6.1/jinns.egg-info/requires.txt +0 -12
  129. jinns-1.6.1/tests/sharding_tests/test_Burgers_x32_multiple_shardings.py +0 -231
  130. {jinns-1.6.1 → jinns-1.7.1}/.gitlab-ci.yml +0 -0
  131. {jinns-1.6.1 → jinns-1.7.1}/AUTHORS +0 -0
  132. {jinns-1.6.1 → jinns-1.7.1}/LICENSE +0 -0
  133. {jinns-1.6.1 → jinns-1.7.1}/Notebooks/ODE/MS_model_Verhulst.ipynb +0 -0
  134. {jinns-1.6.1 → jinns-1.7.1}/Notebooks/ODE/sbinn_data/glucose.dat +0 -0
  135. {jinns-1.6.1 → jinns-1.7.1}/Notebooks/ODE/sbinn_data/meal.dat +0 -0
  136. {jinns-1.6.1 → jinns-1.7.1}/Notebooks/PDE/2d_nonstatio_ou_standardsampling.png +0 -0
  137. {jinns-1.6.1 → jinns-1.7.1}/Notebooks/PDE/OU_1D_nonstatio_solution_grid.npy +0 -0
  138. {jinns-1.6.1 → jinns-1.7.1}/Notebooks/Tutorials/burgers_solution_grid.npy +0 -0
  139. {jinns-1.6.1 → jinns-1.7.1}/codemeta.json +0 -0
  140. {jinns-1.6.1 → jinns-1.7.1}/docs/README.md +0 -0
  141. {jinns-1.6.1 → jinns-1.7.1}/docs/_static/custom_css.css +0 -0
  142. {jinns-1.6.1 → jinns-1.7.1}/docs/_static/favicon.png +0 -0
  143. {jinns-1.6.1 → jinns-1.7.1}/docs/api/advanced/derivative_keys.md +0 -0
  144. {jinns-1.6.1 → jinns-1.7.1}/docs/api/advanced/differential_operators.md +0 -0
  145. {jinns-1.6.1 → jinns-1.7.1}/docs/api/advanced/loss_weight_updates.md +0 -0
  146. {jinns-1.6.1 → jinns-1.7.1}/docs/api/datagenerators/datagenerators_core.md +0 -0
  147. {jinns-1.6.1 → jinns-1.7.1}/docs/api/datagenerators/datagenerators_other.md +0 -0
  148. {jinns-1.6.1 → jinns-1.7.1}/docs/api/loss/dynamic_loss.md +0 -0
  149. {jinns-1.6.1 → jinns-1.7.1}/docs/api/loss/loss_weights.md +0 -0
  150. {jinns-1.6.1 → jinns-1.7.1}/docs/api/loss/loss_xde.md +0 -0
  151. {jinns-1.6.1 → jinns-1.7.1}/docs/api/pinn/hyperpinn.md +0 -0
  152. {jinns-1.6.1 → jinns-1.7.1}/docs/api/pinn/pinn.md +0 -0
  153. {jinns-1.6.1 → jinns-1.7.1}/docs/api/pinn/ppinn.md +0 -0
  154. {jinns-1.6.1 → jinns-1.7.1}/docs/api/pinn/save_load.md +0 -0
  155. {jinns-1.6.1 → jinns-1.7.1}/docs/api/pinn/spinn.md +0 -0
  156. {jinns-1.6.1 → jinns-1.7.1}/docs/api/plot.md +0 -0
  157. {jinns-1.6.1 → jinns-1.7.1}/docs/doc_requirements.txt +0 -0
  158. {jinns-1.6.1 → jinns-1.7.1}/docs/javascripts/katex.js +0 -0
  159. {jinns-1.6.1 → jinns-1.7.1}/docs/maths/fokker_planck.md +0 -0
  160. {jinns-1.6.1 → jinns-1.7.1}/docs/maths/introduction_to_pinns.md +0 -0
  161. {jinns-1.6.1 → jinns-1.7.1}/jinns/data/_AbstractDataGenerator.py +0 -0
  162. {jinns-1.6.1 → jinns-1.7.1}/jinns/data/_CubicMeshPDENonStatio.py +0 -0
  163. {jinns-1.6.1 → jinns-1.7.1}/jinns/data/_CubicMeshPDEStatio.py +0 -0
  164. {jinns-1.6.1 → jinns-1.7.1}/jinns/data/_DataGeneratorParameter.py +0 -0
  165. {jinns-1.6.1 → jinns-1.7.1}/jinns/data/__init__.py +0 -0
  166. {jinns-1.6.1 → jinns-1.7.1}/jinns/data/_utils.py +0 -0
  167. {jinns-1.6.1 → jinns-1.7.1}/jinns/experimental/__init__.py +0 -0
  168. {jinns-1.6.1 → jinns-1.7.1}/jinns/experimental/_diffrax_solver.py +0 -0
  169. {jinns-1.6.1 → jinns-1.7.1}/jinns/loss/_DynamicLoss.py +0 -0
  170. {jinns-1.6.1 → jinns-1.7.1}/jinns/loss/__init__.py +0 -0
  171. {jinns-1.6.1 → jinns-1.7.1}/jinns/loss/_loss_components.py +0 -0
  172. {jinns-1.6.1 → jinns-1.7.1}/jinns/nn/__init__.py +0 -0
  173. {jinns-1.6.1 → jinns-1.7.1}/jinns/nn/_save_load.py +0 -0
  174. {jinns-1.6.1 → jinns-1.7.1}/jinns/nn/_utils.py +0 -0
  175. {jinns-1.6.1 → jinns-1.7.1}/jinns/parameters/__init__.py +0 -0
  176. {jinns-1.6.1 → jinns-1.7.1}/jinns/plot/__init__.py +0 -0
  177. {jinns-1.6.1 → jinns-1.7.1}/jinns/plot/_plot.py +0 -0
  178. {jinns-1.6.1 → jinns-1.7.1}/jinns/solver/__init__.py +0 -0
  179. {jinns-1.6.1 → jinns-1.7.1}/jinns/utils/_ItemizableModule.py +0 -0
  180. {jinns-1.6.1 → jinns-1.7.1}/jinns/utils/__init__.py +0 -0
  181. {jinns-1.6.1 → jinns-1.7.1}/jinns/utils/_utils.py +0 -0
  182. {jinns-1.6.1 → jinns-1.7.1}/jinns/validation/__init__.py +0 -0
  183. {jinns-1.6.1 → jinns-1.7.1}/jinns/validation/_validation.py +0 -0
  184. {jinns-1.6.1 → jinns-1.7.1}/jinns.egg-info/dependency_links.txt +0 -0
  185. {jinns-1.6.1 → jinns-1.7.1}/jinns.egg-info/top_level.txt +0 -0
  186. {jinns-1.6.1 → jinns-1.7.1}/setup.cfg +0 -0
  187. {jinns-1.6.1 → jinns-1.7.1}/tests/conftest.py +0 -0
  188. {jinns-1.6.1 → jinns-1.7.1}/tests/dataGenerator_tests/test_CubicMeshPDENonStatio.py +0 -0
  189. {jinns-1.6.1 → jinns-1.7.1}/tests/dataGenerator_tests/test_CubicMeshPDEStatio.py +0 -0
  190. {jinns-1.6.1 → jinns-1.7.1}/tests/dataGenerator_tests/test_DataGeneratorODE.py +0 -0
  191. {jinns-1.6.1 → jinns-1.7.1}/tests/dataGenerator_tests/test_DataGeneratorParameter.py +0 -0
  192. {jinns-1.6.1 → jinns-1.7.1}/tests/dataGenerator_tests/test_sobol_method.py +0 -0
  193. {jinns-1.6.1 → jinns-1.7.1}/tests/parameters_tests/test_params.py +0 -0
  194. {jinns-1.6.1 → jinns-1.7.1}/tests/pytest.ini +0 -0
  195. {jinns-1.6.1 → jinns-1.7.1}/tests/utils_tests/test_solver_utils.py +0 -0
  196. {jinns-1.6.1 → jinns-1.7.1}/tests/utils_tests/test_subtract_with_check.py +0 -0
{jinns-1.6.1 → jinns-1.7.1}/.gitignore RENAMED
@@ -34,3 +34,7 @@ site/
34
34
  # Coverage
35
35
  .coverage
36
36
  tests/.coverage
37
+
38
+ # uv users
39
+ uv.lock
40
+ .python-version
jinns-1.7.1/.pre-commit-config.yaml ADDED
@@ -0,0 +1,25 @@
1
+ default_stages: [pre-commit]
2
+ repos:
3
+ - repo: https://github.com/astral-sh/ruff-pre-commit
4
+ # Ruff version.
5
+ rev: v0.11.8
6
+ hooks:
7
+ # Run the linter.
8
+ - id: ruff
9
+ args: [ --fix ]
10
+ # Run the formatter.
11
+ - id: ruff-format
12
+ - repo: local
13
+ hooks:
14
+ - id: pylint
15
+ name: pylint
16
+ entry: pylint
17
+ language: system
18
+ types: [python]
19
+ require_serial: true
20
+ args:
21
+ [
22
+ "-rn", # Only display messages
23
+ "-sn", # Don't display the score
24
+ "--errors-only",
25
+ ]