jinns 0.8.3__tar.gz → 0.8.5__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 (106) hide show
  1. {jinns-0.8.3 → jinns-0.8.5}/PKG-INFO +1 -1
  2. {jinns-0.8.3 → jinns-0.8.5}/doc/source/index.rst +8 -0
  3. {jinns-0.8.3 → jinns-0.8.5}/jinns/loss/_DynamicLossAbstract.py +1 -1
  4. {jinns-0.8.3 → jinns-0.8.5}/jinns/solver/_solve.py +3 -1
  5. {jinns-0.8.3 → jinns-0.8.5}/jinns.egg-info/PKG-INFO +1 -1
  6. {jinns-0.8.3 → jinns-0.8.5}/tests/runtests.sh +0 -0
  7. {jinns-0.8.3 → jinns-0.8.5}/.gitignore +0 -0
  8. {jinns-0.8.3 → jinns-0.8.5}/.gitlab-ci.yml +0 -0
  9. {jinns-0.8.3 → jinns-0.8.5}/.pre-commit-config.yaml +0 -0
  10. {jinns-0.8.3 → jinns-0.8.5}/LICENSE +0 -0
  11. {jinns-0.8.3 → jinns-0.8.5}/Notebooks/ODE/1D_Generalized_Lotka_Volterra.ipynb +0 -0
  12. {jinns-0.8.3 → jinns-0.8.5}/Notebooks/ODE/1D_Generalized_Lotka_Volterra_seq2seq.ipynb +0 -0
  13. {jinns-0.8.3 → jinns-0.8.5}/Notebooks/ODE/sbinn_data/glucose.dat +0 -0
  14. {jinns-0.8.3 → jinns-0.8.5}/Notebooks/ODE/sbinn_data/meal.dat +0 -0
  15. {jinns-0.8.3 → jinns-0.8.5}/Notebooks/ODE/systems_biology_informed_neural_network.ipynb +0 -0
  16. {jinns-0.8.3 → jinns-0.8.5}/Notebooks/PDE/1D_non_stationary_Burger.ipynb +0 -0
  17. {jinns-0.8.3 → jinns-0.8.5}/Notebooks/PDE/1D_non_stationary_Fisher_KPP_Bounded_Domain.ipynb +0 -0
  18. {jinns-0.8.3 → jinns-0.8.5}/Notebooks/PDE/2D_Heat_inverse_problem.ipynb +0 -0
  19. {jinns-0.8.3 → jinns-0.8.5}/Notebooks/PDE/2D_Navier_Stokes_PipeFlow.ipynb +0 -0
  20. {jinns-0.8.3 → jinns-0.8.5}/Notebooks/PDE/2D_Navier_Stokes_PipeFlow_Metamodel_hyperpinn.ipynb +0 -0
  21. {jinns-0.8.3 → jinns-0.8.5}/Notebooks/PDE/2D_Navier_Stokes_PipeFlow_SoftConstraints.ipynb +0 -0
  22. {jinns-0.8.3 → jinns-0.8.5}/Notebooks/PDE/2D_Poisson_inverse_problem.ipynb +0 -0
  23. {jinns-0.8.3 → jinns-0.8.5}/Notebooks/PDE/2D_non_stationary_OU.ipynb +0 -0
  24. {jinns-0.8.3 → jinns-0.8.5}/Notebooks/PDE/2d_nonstatio_ou_standardsampling.png +0 -0
  25. {jinns-0.8.3 → jinns-0.8.5}/Notebooks/PDE/OU_1D_nonstatio_solution_grid.npy +0 -0
  26. {jinns-0.8.3 → jinns-0.8.5}/Notebooks/PDE/Reaction_Diffusion_2D_heterogenous_model.ipynb +0 -0
  27. {jinns-0.8.3 → jinns-0.8.5}/Notebooks/PDE/Reaction_Diffusion_2D_homogeneous_metamodel_hyperpinn_diffrax.ipynb +0 -0
  28. {jinns-0.8.3 → jinns-0.8.5}/Notebooks/PDE/burger_solution_grid.npy +0 -0
  29. {jinns-0.8.3 → jinns-0.8.5}/Notebooks/PDE/imperfect_modeling_sobolev_reg.ipynb +0 -0
  30. {jinns-0.8.3 → jinns-0.8.5}/Notebooks/Tutorials/implementing_your_own_PDE_problem.ipynb +0 -0
  31. {jinns-0.8.3 → jinns-0.8.5}/Notebooks/Tutorials/load_save_model.ipynb +0 -0
  32. {jinns-0.8.3 → jinns-0.8.5}/README.md +0 -0
  33. {jinns-0.8.3 → jinns-0.8.5}/doc/Makefile +0 -0
  34. {jinns-0.8.3 → jinns-0.8.5}/doc/source/boundary_conditions.rst +0 -0
  35. {jinns-0.8.3 → jinns-0.8.5}/doc/source/conf.py +0 -0
  36. {jinns-0.8.3 → jinns-0.8.5}/doc/source/data.rst +0 -0
  37. {jinns-0.8.3 → jinns-0.8.5}/doc/source/dynamic_loss.rst +0 -0
  38. {jinns-0.8.3 → jinns-0.8.5}/doc/source/experimental.rst +0 -0
  39. {jinns-0.8.3 → jinns-0.8.5}/doc/source/fokker_planck.qmd +0 -0
  40. {jinns-0.8.3 → jinns-0.8.5}/doc/source/loss.rst +0 -0
  41. {jinns-0.8.3 → jinns-0.8.5}/doc/source/loss_ode.rst +0 -0
  42. {jinns-0.8.3 → jinns-0.8.5}/doc/source/loss_pde.rst +0 -0
  43. {jinns-0.8.3 → jinns-0.8.5}/doc/source/losses.rst +0 -0
  44. {jinns-0.8.3 → jinns-0.8.5}/doc/source/math_pinn.qmd +0 -0
  45. {jinns-0.8.3 → jinns-0.8.5}/doc/source/operators.rst +0 -0
  46. {jinns-0.8.3 → jinns-0.8.5}/doc/source/param_estim_pinn.qmd +0 -0
  47. {jinns-0.8.3 → jinns-0.8.5}/doc/source/rar.rst +0 -0
  48. {jinns-0.8.3 → jinns-0.8.5}/doc/source/seq2seq.rst +0 -0
  49. {jinns-0.8.3 → jinns-0.8.5}/doc/source/solve.rst +0 -0
  50. {jinns-0.8.3 → jinns-0.8.5}/doc/source/solver.rst +0 -0
  51. {jinns-0.8.3 → jinns-0.8.5}/doc/source/utils.rst +0 -0
  52. {jinns-0.8.3 → jinns-0.8.5}/jinns/__init__.py +0 -0
  53. {jinns-0.8.3 → jinns-0.8.5}/jinns/data/_DataGenerators.py +0 -0
  54. {jinns-0.8.3 → jinns-0.8.5}/jinns/data/__init__.py +0 -0
  55. {jinns-0.8.3 → jinns-0.8.5}/jinns/data/_display.py +0 -0
  56. {jinns-0.8.3 → jinns-0.8.5}/jinns/experimental/__init__.py +0 -0
  57. {jinns-0.8.3 → jinns-0.8.5}/jinns/experimental/_diffrax_solver.py +0 -0
  58. {jinns-0.8.3 → jinns-0.8.5}/jinns/loss/_DynamicLoss.py +0 -0
  59. {jinns-0.8.3 → jinns-0.8.5}/jinns/loss/_LossODE.py +0 -0
  60. {jinns-0.8.3 → jinns-0.8.5}/jinns/loss/_LossPDE.py +0 -0
  61. {jinns-0.8.3 → jinns-0.8.5}/jinns/loss/_Losses.py +0 -0
  62. {jinns-0.8.3 → jinns-0.8.5}/jinns/loss/__init__.py +0 -0
  63. {jinns-0.8.3 → jinns-0.8.5}/jinns/loss/_boundary_conditions.py +0 -0
  64. {jinns-0.8.3 → jinns-0.8.5}/jinns/loss/_operators.py +0 -0
  65. {jinns-0.8.3 → jinns-0.8.5}/jinns/solver/__init__.py +0 -0
  66. {jinns-0.8.3 → jinns-0.8.5}/jinns/solver/_rar.py +0 -0
  67. {jinns-0.8.3 → jinns-0.8.5}/jinns/solver/_seq2seq.py +0 -0
  68. {jinns-0.8.3 → jinns-0.8.5}/jinns/utils/__init__.py +0 -0
  69. {jinns-0.8.3 → jinns-0.8.5}/jinns/utils/_hyperpinn.py +0 -0
  70. {jinns-0.8.3 → jinns-0.8.5}/jinns/utils/_optim.py +0 -0
  71. {jinns-0.8.3 → jinns-0.8.5}/jinns/utils/_pinn.py +0 -0
  72. {jinns-0.8.3 → jinns-0.8.5}/jinns/utils/_save_load.py +0 -0
  73. {jinns-0.8.3 → jinns-0.8.5}/jinns/utils/_spinn.py +0 -0
  74. {jinns-0.8.3 → jinns-0.8.5}/jinns/utils/_utils.py +0 -0
  75. {jinns-0.8.3 → jinns-0.8.5}/jinns/utils/_utils_uspinn.py +0 -0
  76. {jinns-0.8.3 → jinns-0.8.5}/jinns.egg-info/SOURCES.txt +0 -0
  77. {jinns-0.8.3 → jinns-0.8.5}/jinns.egg-info/dependency_links.txt +0 -0
  78. {jinns-0.8.3 → jinns-0.8.5}/jinns.egg-info/requires.txt +0 -0
  79. {jinns-0.8.3 → jinns-0.8.5}/jinns.egg-info/top_level.txt +0 -0
  80. {jinns-0.8.3 → jinns-0.8.5}/pyproject.toml +0 -0
  81. {jinns-0.8.3 → jinns-0.8.5}/setup.cfg +0 -0
  82. {jinns-0.8.3 → jinns-0.8.5}/tests/conftest.py +0 -0
  83. {jinns-0.8.3 → jinns-0.8.5}/tests/dataGenerator_tests/test_CubicMeshPDENonStatio.py +0 -0
  84. {jinns-0.8.3 → jinns-0.8.5}/tests/dataGenerator_tests/test_CubicMeshPDEStatio.py +0 -0
  85. {jinns-0.8.3 → jinns-0.8.5}/tests/dataGenerator_tests/test_DataGeneratorODE.py +0 -0
  86. {jinns-0.8.3 → jinns-0.8.5}/tests/dataGenerator_tests/test_DataGeneratorParameter.py +0 -0
  87. {jinns-0.8.3 → jinns-0.8.5}/tests/save_load_tests/test_saving_loading_hyperpinn.py +0 -0
  88. {jinns-0.8.3 → jinns-0.8.5}/tests/save_load_tests/test_saving_loading_pinn.py +0 -0
  89. {jinns-0.8.3 → jinns-0.8.5}/tests/save_load_tests/test_saving_loading_spinn.py +0 -0
  90. {jinns-0.8.3 → jinns-0.8.5}/tests/sharding_tests/test_Burger_x32_multiple_shardings.py +0 -0
  91. {jinns-0.8.3 → jinns-0.8.5}/tests/sharding_tests/test_imperfect_sobolev_x32_multiple_shardings.py +0 -0
  92. {jinns-0.8.3 → jinns-0.8.5}/tests/solver_tests/test_Burger_x32.py +0 -0
  93. {jinns-0.8.3 → jinns-0.8.5}/tests/solver_tests/test_Burger_x64.py +0 -0
  94. {jinns-0.8.3 → jinns-0.8.5}/tests/solver_tests/test_Fisher_x32.py +0 -0
  95. {jinns-0.8.3 → jinns-0.8.5}/tests/solver_tests/test_Fisher_x64.py +0 -0
  96. {jinns-0.8.3 → jinns-0.8.5}/tests/solver_tests/test_GLV_x32.py +0 -0
  97. {jinns-0.8.3 → jinns-0.8.5}/tests/solver_tests/test_GLV_x64.py +0 -0
  98. {jinns-0.8.3 → jinns-0.8.5}/tests/solver_tests/test_NSPipeFlow_x32.py +0 -0
  99. {jinns-0.8.3 → jinns-0.8.5}/tests/solver_tests/test_NSPipeFlow_x64.py +0 -0
  100. {jinns-0.8.3 → jinns-0.8.5}/tests/solver_tests/test_OU2D_x32.py +0 -0
  101. {jinns-0.8.3 → jinns-0.8.5}/tests/solver_tests/test_imperfect_sobolev_x32.py +0 -0
  102. {jinns-0.8.3 → jinns-0.8.5}/tests/solver_tests_spinn/test_Burger_x32.py +0 -0
  103. {jinns-0.8.3 → jinns-0.8.5}/tests/solver_tests_spinn/test_Fisher_x32.py +0 -0
  104. {jinns-0.8.3 → jinns-0.8.5}/tests/solver_tests_spinn/test_NSPipeFlow_x32_spinn.py +0 -0
  105. {jinns-0.8.3 → jinns-0.8.5}/tests/solver_tests_spinn/test_OU2D_x32.py +0 -0
  106. {jinns-0.8.3 → jinns-0.8.5}/tests/solver_tests_spinn/test_ReactionDiffusion_nonhomo_x64.py +0 -0
{jinns-0.8.3 → jinns-0.8.5}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: jinns
3
- Version: 0.8.3
3
+ Version: 0.8.5
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>
{jinns-0.8.3 → jinns-0.8.5}/doc/source/index.rst RENAMED
@@ -8,6 +8,14 @@ Welcome to jinn's documentation!
8
8
 
9
9
  Changelog:
10
10
 
11
+ * v0.8.5:
12
+
13
+ - Merge [!36](https://gitlab.com/mia_jinns/jinns/-/merge_requests/36)
14
+
15
+ * v0.8.4:
16
+
17
+ - Fix a bug: wrong argument in the wrapper function for heterogeneous parameter evaluation of a PDEStatio
18
+
11
19
  * v0.8.3:
12
20
 
13
21
  - Add the possibility to load user-provided tables of parameters in DataGeneratorParameter and not only to randomly sample them.
{jinns-0.8.3 → jinns-0.8.5}/jinns/loss/_DynamicLossAbstract.py RENAMED
@@ -154,7 +154,7 @@ class PDEStatio(DynamicLoss):
154
154
  _params = {
155
155
  "nn_params": params["nn_params"],
156
156
  "eq_params": self.eval_heterogeneous_parameters(
157
- t, u, params, self.eq_params_heterogeneity
157
+ x, u, params, self.eq_params_heterogeneity
158
158
  ),
159
159
  }
160
160
  new_args = args[:-1] + (_params,)
{jinns-0.8.3 → jinns-0.8.5}/jinns/solver/_solve.py RENAMED
@@ -218,6 +218,7 @@ def solve(
218
218
  batch, data, param_data, obs_data = get_batch(data, param_data, obs_data)
219
219
 
220
220
  (
221
+ loss,
221
222
  loss_val,
222
223
  loss_terms,
223
224
  params,
@@ -311,7 +312,7 @@ def solve(
311
312
  )
312
313
 
313
314
 
314
- @partial(jit, static_argnames=["loss", "optimizer"])
315
+ @partial(jit, static_argnames=["optimizer"])
315
316
  def gradient_step(loss, optimizer, batch, params, opt_state, last_non_nan_params):
316
317
  """
317
318
  loss and optimizer cannot be jit-ted.
@@ -330,6 +331,7 @@ def gradient_step(loss, optimizer, batch, params, opt_state, last_non_nan_params
330
331
  )
331
332
 
332
333
  return (
334
+ loss,
333
335
  loss_val,
334
336
  loss_terms,
335
337
  params,
{jinns-0.8.3 → jinns-0.8.5}/jinns.egg-info/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: jinns
3
- Version: 0.8.3
3
+ Version: 0.8.5
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>
{jinns-0.8.3 → jinns-0.8.5}/tests/runtests.sh RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/.gitignore RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/.gitlab-ci.yml RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/.pre-commit-config.yaml RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/LICENSE RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/README.md RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/doc/Makefile RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/doc/source/conf.py RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/doc/source/data.rst RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/doc/source/dynamic_loss.rst RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/doc/source/experimental.rst RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/doc/source/loss.rst RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/doc/source/loss_ode.rst RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/doc/source/loss_pde.rst RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/doc/source/losses.rst RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/doc/source/math_pinn.qmd RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/doc/source/operators.rst RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/doc/source/rar.rst RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/doc/source/seq2seq.rst RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/doc/source/solve.rst RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/doc/source/solver.rst RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/doc/source/utils.rst RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/jinns/__init__.py RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/jinns/data/__init__.py RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/jinns/data/_display.py RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/jinns/loss/_DynamicLoss.py RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/jinns/loss/_LossODE.py RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/jinns/loss/_LossPDE.py RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/jinns/loss/_Losses.py RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/jinns/loss/__init__.py RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/jinns/loss/_operators.py RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/jinns/solver/__init__.py RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/jinns/solver/_rar.py RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/jinns/solver/_seq2seq.py RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/jinns/utils/__init__.py RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/jinns/utils/_hyperpinn.py RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/jinns/utils/_optim.py RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/jinns/utils/_pinn.py RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/jinns/utils/_save_load.py RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/jinns/utils/_spinn.py RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/jinns/utils/_utils.py RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/jinns.egg-info/SOURCES.txt RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/jinns.egg-info/requires.txt RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/pyproject.toml RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/setup.cfg RENAMED
File without changes
{jinns-0.8.3 → jinns-0.8.5}/tests/conftest.py RENAMED
File without changes