fluxfem 0.1.3__py3-none-any.whl → 0.2.0__py3-none-any.whl

fluxfem-0.2.0.dist-info/METADATA

@@ -0,0 +1,303 @@
+Metadata-Version: 2.1
+Name: fluxfem
+Version: 0.2.0
+Summary: FluxFEM: A weak-form-centric differentiable finite element framework in JAX
+License: Apache-2.0
+Author: Kohei Watanabe
+Author-email: koheitech001@gmail.com
+Requires-Python: >=3.12,<3.14
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.12
+Provides-Extra: cpu
+Provides-Extra: cuda12
+Provides-Extra: petsc
+Requires-Dist: jax (>=0.8.2,<0.9.0) ; extra == "cpu" or extra == "cuda12"
+Requires-Dist: jax-cuda12-pjrt (==0.8.2) ; extra == "cuda12"
+Requires-Dist: jax-cuda12-plugin (==0.8.2) ; extra == "cuda12"
+Requires-Dist: jaxlib (>=0.8.2,<0.9.0) ; extra == "cpu" or extra == "cuda12"
+Requires-Dist: matplotlib (>=3.10.7,<4.0.0)
+Requires-Dist: meshio (>=5.3.5,<6.0.0)
+Requires-Dist: petsc4py (==3.23.6) ; extra == "petsc"
+Requires-Dist: pyproject (>=1!0.1.2,<1!0.2.0)
+Requires-Dist: pyproject-toml (>=0.1.0,<0.2.0)
+Requires-Dist: pyvista (>=0.46.4,<0.47.0)
+Description-Content-Type: text/markdown
+
+[![PyPI version](https://img.shields.io/pypi/v/fluxfem.svg?cacheSeconds=60)](https://pypi.org/project/fluxfem/)
+[![License: Apache-2.0](https://img.shields.io/badge/License-Apache%202.0-blue.svg)](https://opensource.org/licenses/Apache-2.0)
+[![Python Version](https://img.shields.io/pypi/pyversions/fluxfem.svg)](https://pypi.org/project/fluxfem/)
+![CI](https://github.com/kevin-tofu/fluxfem/actions/workflows/python-tests.yml/badge.svg)
+![CI](https://github.com/kevin-tofu/fluxfem/actions/workflows/sphinx.yml/badge.svg)
+[![DOI](https://zenodo.org/badge/DOI/10.5281/zenodo.18055465.svg)](https://doi.org/10.5281/zenodo.18055465)
+
+
+# FluxFEM
+A weak-form-centric differentiable finite element framework in JAX,
+where variational forms are treated as first-class, differentiable programs.
+
+## Examples and Features
+<table>
+  <tr>
+    <td align="center"><b>Example 1: Diffusion</b></td>
+    <td align="center"><b>Example 2: Neo Neohookean Hyper Elasticity</b></td>
+  </tr>
+  <tr>
+    <td align="center">
+      <img src="https://media.githubusercontent.com/media/kevin-tofu/fluxfem/main/assets/diffusion_mms_timeseries.gif" alt="Diffusion-mms" width="400">
+    </td>
+    <td align="center">
+      <img src="https://media.githubusercontent.com/media/kevin-tofu/fluxfem/main/assets/Neo-Hookean-deformedx20000.png" alt="Neo-Hookean" width="400">
+    </td>
+  </tr>
+</table>
+
+
+## Features
+- Built on JAX, enabling automatic differentiation with grad, jit, vmap, and related transformations.
+- Weak-form–centric API that keeps formulations close to code; weak forms are represented as expression trees and compiled into element kernels, enabling automatic differentiation of residuals, tangents, and objectives.
+- Two assembly approaches: tensor-based (scikit-fem–style) assembly and weak-form-based assembly.
+- Handles both linear and nonlinear analyses with AD in JAX.
+- Optional PETSc/PETSc-shell solvers via `petsc4py` for scalable linear solves (add `fluxfem[petsc]`).
+
+## Usage 
+
+This library provides two assembly approaches.
+
+- A tensor-based assembly, where trial and test functions are represented explicitly as element-level tensors and assembled accordingly (in the style of scikit-fem).
+- A weak-form-based assembly, where the variational form is written symbolically and compiled before assembly.
+
+The two approaches are functionally equivalent and share the same element-level execution model,
+but they differ in how you author the weak form. The example below mirrors the paper's diffusion
+case and makes the distinction explicit with `jnp`.
+
+
+## Assembly Flow
+All expressions are first compiled into an element-level evaluation plan,
+which operates on quadrature-point–major tensors.
+This plan is then executed independently for each element during assembly.
+
+As a result, both assembly approaches:
+- use the same quadrature-major (q, a, i) data layout,
+- perform element-local tensor contractions,
+- and are fully compatible with JAX transformations such as `jit`, `vmap`, and automatic differentiation.
+
+### kernel-based assembly (explicit JIT units)
+If you want to control JIT boundaries explicitly, build a JIT-compiled element kernel
+and pass it to `space.assemble`. The kernel must return the integrated element
+contribution (not the quadrature integrand). For untagged raw kernels, pass `kind=`.
+
+```Python
+import fluxfem as ff
+import jax
+import jax.numpy as jnp
+
+space = ff.make_hex_space(mesh, dim=1, intorder=2)
+
+   # bilinear: kernel(ctx) -> (n_ldofs, n_ldofs)
+   ker_K = ff.make_element_bilinear_kernel(ff.diffusion_form, 1.0, jit=True)
+   K = space.assemble(ff.diffusion_form, 1.0, kernel=ker_K)
+
+# linear: kernel(ctx) -> (n_ldofs,)
+def linear_kernel(ctx):
+    integrand = ff.scalar_body_force_form(ctx, 2.0)
+    wJ = ctx.w * ctx.test.detJ
+    return (integrand * wJ[:, None]).sum(axis=0)
+
+   ker_F = jax.jit(linear_kernel)
+   F = space.assemble(ff.scalar_body_force_form, 2.0, kernel=ker_F)
+```
+
+### tensor-based vs weak-form-based (diffusion example)
+
+#### tensor-based assembly
+The tensor-based assembly provides an explicit, low-level formulation with element kernels written using jax.numpy.(`jnp`).
+```Python
+import fluxfem as ff
+import jax.numpy as jnp
+
+@ff.kernel(kind="bilinear", domain="volume")
+def diffusion_form(ctx: ff.FormContext, kappa):
+    # ctx.test.gradN / ctx.trial.gradN: (n_qp, n_nodes, dim)
+    # output tensor: (n_qp, n_nodes, n_nodes)
+    return kappa * jnp.einsum("qia,qja->qij", ctx.test.gradN, ctx.trial.gradN)
+
+space = ff.make_hex_space(mesh, dim=3, intorder=2)
+params = ff.Params(kappa=1.0)
+K_ts = space.assemble(diffusion_form, params=params.kappa)
+```
+
+#### weak-form-based assembly
+In the weak-form-based assembly, the variational formulation itself is the primary object.
+The expression below defines a symbolic computation graph, which is later compiled and executed at the element level.
+
+```Python
+import fluxfem as ff
+import fluxfem.helpers_wf as h_wf
+
+space = ff.make_hex_space(mesh, dim=3, intorder=2)
+params = ff.Params(kappa=1.0)
+
+# u, v are symbolic trial/test fields (weak-form DSL objects).
+# u.grad / v.grad are symbolic nodes (expression tree), not numeric arrays.
+# dOmega() is the integral measure; the whole expression is compiled before assembly.
+form_wf = ff.BilinearForm.volume(
+    lambda u, v, p: p.kappa * (v.grad @ u.grad) * h_wf.dOmega()
+).get_compiled()
+
+K_wf = space.assemble(form_wf, params=params)
+```
+
+### Linear Elasticity assembly (weak-form based assembly)
+
+```Python
+import fluxfem as ff
+import fluxfem.helpers_wf as h_wf
+
+space = ff.make_hex_space(mesh, dim=3, intorder=2)
+D = ff.isotropic_3d_D(1.0, 0.3)
+
+form_wf = ff.BilinearForm.volume(
+    lambda u, v, D: h_wf.ddot(v.sym_grad, D @ u.sym_grad) * h_wf.dOmega()
+).get_compiled()
+
+K = space.assemble(form_wf, params=D)
+```
+
+### Neo-Hookean residual assembly (weak-form DSL)
+Below is a Neo-Hookean hyperelasticity example written in weak form.
+The residual is expressed symbolically and compiled into element-level kernels executed per element.
+No manual derivation of tangent operators is required; consistent tangents (Jacobians) for Newton-type solvers are obtained automatically via JAX AD.
+
+```Python
+def neo_hookean_residual_wf(v, u, params):
+    mu = params["mu"]
+    lam = params["lam"]
+    F = h_wf.I(3) + h_wf.grad(u)  # deformation gradient
+    C = h_wf.matmul(h_wf.transpose(F), F)
+    C_inv = h_wf.inv(C)
+    J = h_wf.det(F)
+
+    S = mu * (h_wf.I(3) - C_inv) + lam * h_wf.log(J) * C_inv
+    dE = 0.5 * (h_wf.matmul(h_wf.grad(v), F) + h_wf.transpose(h_wf.matmul(h_wf.grad(v), F)))
+    return h_wf.ddot(S, dE) * h_wf.dOmega()
+
+  res_form = ff.ResidualForm.volume(neo_hookean_residual_wf).get_compiled()
+```
+
+
+### autodiff + jit compile
+
+You can differentiate through the solve and JIT compile the hot path.
+The inverse diffusion tutorial shows this pattern:
+
+```Python
+def loss_theta(theta):
+    kappa = jnp.exp(theta)
+    u = solve_u_jit(kappa, traction_true)
+    diff = u[obs_idx_j] - u_obs[obs_idx_j]
+    return 0.5 * jnp.mean(diff * diff)
+
+solve_u_jit = jax.jit(solve_u)
+loss_theta_jit = jax.jit(loss_theta)
+grad_fn = jax.jit(jax.grad(loss_theta))
+```
+
+### Mixed systems
+
+Mixed problems can be assembled from residual blocks and solved as a coupled system.
+
+```Python
+import fluxfem as ff
+import jax.numpy as jnp
+
+mixed = ff.MixedFESpace({"u": space_u, "p": space_p})
+residuals = ff.make_mixed_residuals(
+    u=res_u,  # (v, u, params) -> Expr
+    p=res_p,  # (q, u, params) -> Expr
+)
+problem = ff.MixedProblem(mixed, residuals, params=ff.Params(alpha=1.0))
+
+u0 = jnp.zeros(mixed.n_dofs)
+R = problem.assemble_residual(u0)
+J = problem.assemble_jacobian(u0, return_flux_matrix=True)
+```
+
+### Block assembly
+
+For constraints like contact problems (e.g., adding Lagrange multipliers), build
+a block matrix explicitly:
+
+```Python
+from fluxfem import solver as ff_solver
+
+# Example blocks from contact coupling
+K_uu = ...
+K_cc = ...
+K_uc = ...
+
+blocks = ff_solver.make_block_matrix(
+    diag=ff_solver.block_diag(u=K_uu, c=K_cc),
+    rel={("u", "c"): K_uc},
+    sizes={"u": K_uu.shape[0], "c": K_cc.shape[0]},
+    symmetric=True,
+    transpose_rule="T",
+)
+```
+
+FluxFEM also provides contact utilities like `ContactSurfaceSpace` to build constraint contributions.
+
+
+## Documentation
+
+Full documentation, tutorials, and API reference are hosted at [this site](https://fluxfem.readthedocs.io/en/latest/).
+
+## Tutorials
+
+- `tutorials/linearelastic_tensile_bar.py` (linear elasticity, weak-form assembly)
+- `tutorials/neo_hookean_cantilever.py` (nonlinear hyperelasticity)
+- `tutorials/thermoelastic_bar_1d.py` / `tutorials/thermoelastic_bar_1d_mixed.py` (thermoelastic coupling)
+- `tutorials/petsc_shell_poisson_demo.py` (PETSc shell solver integration; see also `tutorials/petsc_shell_poisson_pmat_demo.py`)
+
+## Setup
+
+You can install **FluxFEM** either via **pip** or **Poetry**.
+
+#### Supported Python Versions
+
+FluxFEM supports **Python 3.11–3.13**:
+
+
+**Choose one of the following methods:**
+
+### Using pip
+```bash
+pip install fluxfem
+pip install "fluxfem[cuda12]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
+```
+
+### Using poetry
+```bash
+poetry add fluxfem
+poetry add fluxfem[cuda12]
+```
+
+## PETSc Integration
+
+Optional PETSc-based solvers are available via `petsc4py`. Enable with the extra:
+
+```bash
+poetry add fluxfem --extras "petsc"
+```
+
+Note: newer `petsc4py` expects PETSc builds that include the `PetscRegressor`
+API. If your PETSc build does not have it, `petsc4py` will fail to compile. In
+that case, rebuild PETSc with regressor support or pin `petsc4py` to a version
+compatible with your PETSc build.
+
+GPU note: this repo currently tests CUDA via the `cuda12` extra only. Other CUDA
+versions are not covered by CI and may require manual JAX installation.
+
+## Acknowledgements
+I acknowledge the open-source software, libraries, and communities that made this work possible.
+

fluxfem-0.2.0.dist-info/RECORD

@@ -0,0 +1,59 @@
+fluxfem/__init__.py,sha256=KPEgeWjheIAHQ28MWazyzLCN-lR0khSRltq06Q9tlUg,8022
+fluxfem/core/__init__.py,sha256=BKhJR8dK8oZkX9yW0W8m7u5nHDhnwmUg0wWF3iL4Hm8,10964
+fluxfem/core/assembly.py,sha256=rBQ3RPEVxfCuCnKc9Xxz6vLrRZ9ROO8JNggfV4uGWTc,46070
+fluxfem/core/basis.py,sha256=bh1evqiS5tNG7bqyq7ZzipQR8SBEIM-B7MzJ6_niK0A,32397
+fluxfem/core/context_types.py,sha256=QGv1oc_9Crn3vztyPy_uUN0TpmDkgH5HPrhYzZ_kuys,663
+fluxfem/core/data.py,sha256=_byAZTasIGGwNRU3gqw_ZZR1HMVLJpk9gix2AI3AJWk,1818
+fluxfem/core/dtypes.py,sha256=FnJ98C3yIibaatN-wmLO3ksI6XabA7G3rI_SOxPmko8,251
+fluxfem/core/forms.py,sha256=uC5WxG3h3G7I9FxNRVvA8JLVpJP-QjNZtj8fjxavqVE,7697
+fluxfem/core/interp.py,sha256=v9HzkYgQnEqf6GnB9tXqfrNk0N7YsmPcZBfNGNUU3b8,2004
+fluxfem/core/mixed_assembly.py,sha256=4lL15AsKV4KA-D_mhwc_zXyIhff1fwMkYj4iPWDbDwk,8513
+fluxfem/core/mixed_space.py,sha256=8LZU2VibMdHP4tq_1wrf1rY1LVZo6zZiw8UE6cPaqus,13285
+fluxfem/core/mixed_weakform.py,sha256=1RizZ1KuyCGse3LDbHwCcMHP0YjVKBrwE_USm7COIoQ,2848
+fluxfem/core/solver.py,sha256=OOKiEsW9j4BPCtwgIYyaUHh0Bw5dAd975aGl2izOYqQ,3772
+fluxfem/core/space.py,sha256=XHs0NamEBVNyvaXU6FdIY9xgeY3VU8jiy8cX8qZnOyk,22159
+fluxfem/core/weakform.py,sha256=Sgov4zXfEtYtUASDcy5g1bZu0XsnBMUtQuw1bwj6Abg,70163
+fluxfem/helpers_ts.py,sha256=T3HCqXniF116lvznq0b7CDUok5_yZZc8l9XtlE81yvg,221
+fluxfem/helpers_wf.py,sha256=pA8bprvbdXZL_qourm07xaJAsGyNnAP8wXmJf8fOj-k,2158
+fluxfem/mesh/__init__.py,sha256=u2CzTjFxj_CRbi18FezIDh73k_lan_3wGBifTkdJH4w,2418
+fluxfem/mesh/base.py,sha256=ghsvH8kvwpprmR6x1HkSG0H6QTOpl0CVk4HCBDSIdPY,22716
+fluxfem/mesh/contact.py,sha256=WDwwB0s_Zbd7qAEx4eLGr9_zn2KKh5SS0tfTOgnSTRU,28921
+fluxfem/mesh/dtypes.py,sha256=FnJ98C3yIibaatN-wmLO3ksI6XabA7G3rI_SOxPmko8,251
+fluxfem/mesh/hex.py,sha256=VhH0THJMHPhNtwAvqBjIaeNMzBUd8SWzkih2FcR1NWc,12455
+fluxfem/mesh/io.py,sha256=Nlg0urKVDluX1r2guMjvPWNL-SlS2lOc4eLKuKLex9s,3044
+fluxfem/mesh/mortar.py,sha256=-OhyHkaFEYCOoIvf8GpNUJB0lMGQe7hBGzEQ3rV7zYk,147865
+fluxfem/mesh/predicate.py,sha256=7O9kcKYg_OPtwhqYVdA-j9LQYtcG2Nl781UupKm6U-A,1505
+fluxfem/mesh/supermesh.py,sha256=DNDE3-Tck9KSWIbOWGJ7Dduuag9ofZF_B7PqjNecjtA,10407
+fluxfem/mesh/surface.py,sha256=-xNhs3pLnU5-a8ct8hQTd8YzvjETgTRA3tYK9Dwb-Fw,9791
+fluxfem/mesh/tet.py,sha256=Q-2S-b3N4CF3i40u5tt5mbPn_fYcvCJx4K6Xiyq11-A,10077
+fluxfem/physics/__init__.py,sha256=gUrVF1FSiSVZezqRGvKSTcZDH79Ty3opiJ9-H1gycqY,1426
+fluxfem/physics/diffusion.py,sha256=T3jIGqnuaoSMB8q9x6qNwY23V55-lQGb6ome_WQnOS8,537
+fluxfem/physics/elasticity/__init__.py,sha256=4pAqAstT9T20pY5q417c9FQAarq_T8g2hpqDtrUDkcw,1061
+fluxfem/physics/elasticity/hyperelastic.py,sha256=TvclMK050_h8cO97b3vx2Br8QOoKYzzDfM4nbUfOfL8,3461
+fluxfem/physics/elasticity/linear.py,sha256=JU_LiQoYbABMrJzvFRHKnnCT44iK4WQKZxRu1GLxyq8,2255
+fluxfem/physics/elasticity/materials.py,sha256=f2y7_nY3KMSmD2-G4znondAOvh-XDvJD_-zcE2EZcKw,977
+fluxfem/physics/elasticity/stress.py,sha256=i3TypnLB-Hhd3UsV2tftOCjF1-KNNpnVEy8rYqT2wIM,1135
+fluxfem/physics/operators.py,sha256=LgEvtcDvhskd7BokVsX_wtzceoI2zrLYyB5HWsUBFg4,3009
+fluxfem/physics/postprocess.py,sha256=7NKcSJO_Himw1WTJgM4YWxf1KkjGuBwojxY_0Vwy9Ow,4418
+fluxfem/solver/__init__.py,sha256=gv57yXua8RSsubZ5SbTaXnKALnRdsKOSuCBR7nuBiCo,2337
+fluxfem/solver/bc.py,sha256=IJa9WmAkeLu73fmxppEZRXfB-1o_B-9tQ8uKXZHdxFU,16649
+fluxfem/solver/block_matrix.py,sha256=qJ6rOtiJyxYlSHkzYbi-T_BN7Xfd0drNzXrM8mdq8uo,4187
+fluxfem/solver/block_system.py,sha256=r5nvt2OWGS9iYq0n2G2HB3MM--ThwGsGoUwcM-w6ZFw,17659
+fluxfem/solver/cg.py,sha256=ZP4OcFeB-froIXcg_PVEobf5BYKz5gONALcEpKCbr_k,11101
+fluxfem/solver/dirichlet.py,sha256=tmKpXWP_8v_jIb337dTkvDxsgkOcoE8ViKuZajQn8L4,14875
+fluxfem/solver/history.py,sha256=HhKSmo2B1GaBYEsrifxQxeaeRMiaqm5f6g6DNz5Qb0U,764
+fluxfem/solver/newton.py,sha256=fVqASbzXrQb9ZqjKol2lWioEr7coi4sWaYMGvYDztqc,23444
+fluxfem/solver/petsc.py,sha256=PjCRomFU3BWjwCO_ZAa8dS5sh_OhXvNSZykURmSroRY,14910
+fluxfem/solver/preconditioner.py,sha256=PdjaHUVgwHGUSO29Od04aQtr24cq9XX9atgClfKKZ-8,3578
+fluxfem/solver/result.py,sha256=DLLyaHVJ9NYotDqFD0IHlhxg9Hbs726bepqCG_59t94,3256
+fluxfem/solver/solve_runner.py,sha256=OuEQlxSIhp9ahaRzS45ZWqgq1ztkf-oDcvp394nGZ7g,30417
+fluxfem/solver/solver.py,sha256=sU9z4btnX5pPUAzxFLRusKNGsGg3pu1njRqfkRjQE9I,5640
+fluxfem/solver/sparse.py,sha256=Znn50ncFDrqoWNOwxIvXmcK5WpDtLy5ktvBuavkAFFU,11875
+fluxfem/tools/__init__.py,sha256=S7flze8kfHABc0fdhGgdELzKO8aeBCrDbwRUE3aTGpA,95
+fluxfem/tools/jit.py,sha256=WVThoNs2XD8ZVCeyLjBgbFaZXHot5RK_RNzvvAjSaew,1110
+fluxfem/tools/timer.py,sha256=fvMrKqjOT2S_DoGlTYC5L5TtBnoLS4zyiAZtJVJGt0o,22848
+fluxfem/tools/visualizer.py,sha256=aDqSgtX9xOu120-zCzJfwlCOUmxjL_kJ6kAvj27-H14,3733
+fluxfem-0.2.0.dist-info/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+fluxfem-0.2.0.dist-info/METADATA,sha256=3RicCvbWhQOUTeQoMgmpYz-q6EznEmYcCn9h63B-Dv0,10931
+fluxfem-0.2.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+fluxfem-0.2.0.dist-info/RECORD,,

fluxfem-0.1.3.dist-info/METADATA

@@ -1,125 +0,0 @@
-Metadata-Version: 2.1
-Name: fluxfem
-Version: 0.1.3
-Summary: FluxFEM: A weak-form-centric differentiable finite element framework in JAX
-License: Apache-2.0
-Author: Kohei Watanabe
-Author-email: koheitech001@gmail.com
-Requires-Python: >=3.11,<3.14
-Classifier: License :: OSI Approved :: Apache Software License
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Requires-Dist: jax (==0.8.2)
-Requires-Dist: jaxlib (==0.8.2)
-Requires-Dist: matplotlib (>=3.10.7,<4.0.0)
-Requires-Dist: meshio (>=5.3.5,<6.0.0)
-Requires-Dist: pyvista (>=0.46.4,<0.47.0)
-Description-Content-Type: text/markdown
-
-[![PyPI version](https://img.shields.io/pypi/v/fluxfem.svg?cacheSeconds=60)](https://pypi.org/project/fluxfem/)
-[![License: Apache-2.0](https://img.shields.io/badge/License-Apache%202.0-blue.svg)](https://opensource.org/licenses/Apache-2.0)
-[![Python Version](https://img.shields.io/pypi/pyversions/fluxfem.svg)](https://pypi.org/project/fluxfem/)
-![CI](https://github.com/kevin-tofu/fluxfem/actions/workflows/python-tests.yml/badge.svg)
-![CI](https://github.com/kevin-tofu/fluxfem/actions/workflows/sphinx.yml/badge.svg)
-
-# FluxFEM
- A weak-form-centric differentiable finite element framework in JAX
-
-## Examples and Features
-<table>
-  <tr>
-    <td align="center"><b>Example 1: Diffusion</b></td>
-    <td align="center"><b>Example 2: Neo Neohookean Hyper Elasticity</b></td>
-  </tr>
-  <tr>
-    <td align="center">
-      <img src="https://media.githubusercontent.com/media/kevin-tofu/fluxfem/main/assets/diffusion_mms_timeseries.gif" alt="Diffusion-mms" width="400">
-    </td>
-    <td align="center">
-      <img src="https://media.githubusercontent.com/media/kevin-tofu/fluxfem/main/assets/Neo-Hookean-deformedx20000.png" alt="Neo-Hookean" width="400">
-    </td>
-  </tr>
-</table>
-
-
-## Features
-- Built on JAX, enabling automatic differentiation and high-performance execution via grad, jit, vmap, and related transformations.
-
-- A FEM framework with a weak-form–centric API, emphasizing a smooth transition from theoretical formulations to practical code implementations.
-
-- Supports two assembly approaches: weak-form-based assembly and a tensor-based (scikit-fem–style) assembly.
-
-- enables to handle both Linear / Non-Linear analysis with AD in JAX
-
-## Usage 
-
-This library provides two assembly approaches.
-
-- A weak-form-based assembly, where the variational form is written and assembled directly.  
-- A tensor-based assembly, where trial and test functions are represented explicitly as tensors and assembled accordingly (in the style of scikit-fem).  
-The first approach offers simplicity and convenience, as mathematical expressions can be written almost directly in code.
-However, for more complex operations, the second approach can be easier to implement in practice.
-This is because the weak-form-based assembly is ultimately transformed into the tensor-based representation internally during computation.
-
-### weak-form-based assembly
-```Python
-import fluxfem as ff
-import fluxfem.helpers_wf as h_wf
-
-space = ff.make_hex_space(mesh, dim=3, intorder=2)
-D = ff.isotropic_3d_D(1.0, 0.3)
-bilinear_form = ff.BilinearForm.volume(
-    lambda u, v, D: h_wf.ddot(v.sym_grad, D @ u.sym_grad) * h_wf.dOmega()
-)
-K_wf = space.assemble_bilinear_form(
-    bilinear_form.get_compiled(),
-    params=D,
-)
-```
-
-### tensor-based assembly (scikit-fem-style)
-
-```Python
-import fluxfem as ff
-import numpy as np
-import fluxfem.helpers_ts as h_ts
-
-def linear_elasticity_form(ctx: ff.FormContext, D: np.ndarray) -> ff.jnp.ndarray:
-        Bu = h_ts.sym_grad(ctx.trial)
-        Bv = h_ts.sym_grad(ctx.test)
-        return h_ts.ddot(Bv, D, Bu)
-
-
-space = ff.make_hex_space(mesh, dim=3, intorder=2)
-D = ff.isotropic_3d_D(1.0, 0.3)
-K = space.assemble_bilinear_form(linear_elasticity_form, params=D)
-```
-
-## Documentation
-
-
-## SetUp
-
-You can install **FluxFEM** either via **pip** or **Poetry**.
-
-#### Supported Python Versions
-
-FluxFEM supports **Python 3.11–3.13**:
-
-
-**Choose one of the following methods:**
-
-### Using pip
-```bash
-pip install fluxfem
-```
-
-### Using poetry
-```bash
-poetry add fluxfem
-```
-
-## Acknowledgements
- I acknoldege everythings that made this work possible.
-

fluxfem-0.1.3.dist-info/RECORD

@@ -1,47 +0,0 @@
-fluxfem/__init__.py,sha256=wpnJOQj4PSIzWfYd6R0WjAAxN6V8CV7GfxeZqM1WNjc,8145
-fluxfem/core/__init__.py,sha256=DgGJ1Z8ZPfEFjv-q_srihazOxqcPMyEWM2w495ojmHQ,7490
-fluxfem/core/assembly.py,sha256=yhc14y4XWvmBpiLun2eH9k1mjAsNmpTbHPJS2VJFO3M,25370
-fluxfem/core/basis.py,sha256=ZxTQUAEC2UBuvxGhiSqAPjVnUwP_nIptY_b9Wn2etak,32113
-fluxfem/core/data.py,sha256=_byAZTasIGGwNRU3gqw_ZZR1HMVLJpk9gix2AI3AJWk,1818
-fluxfem/core/dtypes.py,sha256=WQ_wfJW_R8Mty76mS7h-R2xgCokraEWjsUvjeKYmVJU,118
-fluxfem/core/forms.py,sha256=ZxmpmaZa-uTRi-0Dltvn0VkEcXpW4HgNFr8Po0b8JyQ,7264
-fluxfem/core/interp.py,sha256=v9HzkYgQnEqf6GnB9tXqfrNk0N7YsmPcZBfNGNUU3b8,2004
-fluxfem/core/solver.py,sha256=nNpb90ddZq8Ek6XPerlKzV4YWtC5ZgEcsRv4KeoFIug,3770
-fluxfem/core/space.py,sha256=jj4koa6_7lBBtXWGrF1CehOBKWg--uUW_7q8pB2EFCk,13902
-fluxfem/core/weakform.py,sha256=Q2SoOX1KwwPTigJMDUygQnEuKoxBAHh0R3qwI_sf4gY,27470
-fluxfem/helpers_ts.py,sha256=T3HCqXniF116lvznq0b7CDUok5_yZZc8l9XtlE81yvg,221
-fluxfem/helpers_wf.py,sha256=JUKiA5p_V9NmUdskHYgYYOlsa8xKTZzBkJCoeUqTu1c,574
-fluxfem/mesh/__init__.py,sha256=3ECN0qsVzOnEGTHfrWJ2PV7RJUbSboow1ewfpknI4F4,927
-fluxfem/mesh/base.py,sha256=NCJxqKLb6AOC2oGN-GfjC8Uu1Mddpa3E10RNYCyLYtI,10214
-fluxfem/mesh/hex.py,sha256=37xSQDDIndbz7JfBCSdI9ptV19L4_OZ4XgKfkFRGS00,12373
-fluxfem/mesh/io.py,sha256=ZNTXNRB9_tx_J99y-c46ypboQh1IpTHretWXtYWdCag,2834
-fluxfem/mesh/predicate.py,sha256=7O9kcKYg_OPtwhqYVdA-j9LQYtcG2Nl781UupKm6U-A,1505
-fluxfem/mesh/surface.py,sha256=mmtePb1ptd9BSbD79oyPnQwiKadpyVO41xSHDr2XylE,9176
-fluxfem/mesh/tet.py,sha256=Yjmx_2UsCxO_rl0Jr60au5zccBVWLJvAaGEzhLCi_LA,9768
-fluxfem/physics/__init__.py,sha256=gUrVF1FSiSVZezqRGvKSTcZDH79Ty3opiJ9-H1gycqY,1426
-fluxfem/physics/diffusion.py,sha256=gBRKMuhWGnp7wSlbqYKn2WDKvBrVdtCQKV8-qBabmw4,462
-fluxfem/physics/elasticity/__init__.py,sha256=4pAqAstT9T20pY5q417c9FQAarq_T8g2hpqDtrUDkcw,1061
-fluxfem/physics/elasticity/hyperelastic.py,sha256=uDZIgSeVJ1JgIf5aEFzFyng6qcDHP-seoHNH_aUA4Zg,3364
-fluxfem/physics/elasticity/linear.py,sha256=upiX8Qq5v8J0Y-Lhq8OsIFnwQ_AnJc9nJj0g0vWeNcg,2074
-fluxfem/physics/elasticity/materials.py,sha256=f2y7_nY3KMSmD2-G4znondAOvh-XDvJD_-zcE2EZcKw,977
-fluxfem/physics/elasticity/stress.py,sha256=i3TypnLB-Hhd3UsV2tftOCjF1-KNNpnVEy8rYqT2wIM,1135
-fluxfem/physics/operators.py,sha256=LgEvtcDvhskd7BokVsX_wtzceoI2zrLYyB5HWsUBFg4,3009
-fluxfem/physics/postprocess.py,sha256=7NKcSJO_Himw1WTJgM4YWxf1KkjGuBwojxY_0Vwy9Ow,4418
-fluxfem/solver/__init__.py,sha256=WkQRDWVIxqXuRGrbizl0KVkr7a_yY0jWlsTDI66gaiE,1133
-fluxfem/solver/bc.py,sha256=EYrvQbTMQG2WxHbJYJD2nefxwrpiGS7p8ant6R3APgU,15555
-fluxfem/solver/cg.py,sha256=3fLtK3oUpR7lQtRPPtE64-PHUjeIp3pDuMZZqVdCXfk,8974
-fluxfem/solver/dirichlet.py,sha256=MCXgpJ9lAF7NrK2NeuCxVWgOVfCiPeswIYhbadkwh-Q,3982
-fluxfem/solver/history.py,sha256=HhKSmo2B1GaBYEsrifxQxeaeRMiaqm5f6g6DNz5Qb0U,764
-fluxfem/solver/newton.py,sha256=aeSQ1ET9dc-xU6HJWn6KNrEg0FA49nSXsoaYuW5BTvM,16251
-fluxfem/solver/result.py,sha256=LDtazqofyBmA0pd3zE5eadgKwW053HRcOwJ5k7ClbsA,2357
-fluxfem/solver/solve_runner.py,sha256=MCwmHUKnJCX_mFDPFKNp-_yMbviWln5s56ENWZvOj4c,20974
-fluxfem/solver/solver.py,sha256=kqXz2qW50gyyFQagUePgNiFcgsLVA94H99_8sHZWiyo,5162
-fluxfem/solver/sparse.py,sha256=wyT_XBIMYD2Nrw_ko8FkaPzLWnLm25M41wF2Z2YAbZk,7550
-fluxfem/tools/__init__.py,sha256=S7flze8kfHABc0fdhGgdELzKO8aeBCrDbwRUE3aTGpA,95
-fluxfem/tools/jit.py,sha256=WVThoNs2XD8ZVCeyLjBgbFaZXHot5RK_RNzvvAjSaew,1110
-fluxfem/tools/timer.py,sha256=fvMrKqjOT2S_DoGlTYC5L5TtBnoLS4zyiAZtJVJGt0o,22848
-fluxfem/tools/visualizer.py,sha256=aDqSgtX9xOu120-zCzJfwlCOUmxjL_kJ6kAvj27-H14,3733
-fluxfem-0.1.3.dist-info/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-fluxfem-0.1.3.dist-info/METADATA,sha256=FkdJPsdozjTFWGBp3f87EOaRp2IPdo2G2oW1qPCA4-g,4259
-fluxfem-0.1.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-fluxfem-0.1.3.dist-info/RECORD,,