LICENSE.md
README.md
setup.py
examples/__init__.py
examples/stars/__init__.py
examples/stars/_solve_single_star.py
examples/stars/_starbox.py
examples/stars/solve_stars.py
jax_dips/__init__.py
jax_dips.egg-info/PKG-INFO
jax_dips.egg-info/SOURCES.txt
jax_dips.egg-info/dependency_links.txt
jax_dips.egg-info/not-zip-safe
jax_dips.egg-info/requires.txt
jax_dips.egg-info/top_level.txt
jax_dips/_experimental/__init__.py
jax_dips/_experimental/compositions.py
jax_dips/_jaxmd_modules/__init__.py
jax_dips/_jaxmd_modules/dataclasses.py
jax_dips/_jaxmd_modules/energy.py
jax_dips/_jaxmd_modules/partition.py
jax_dips/_jaxmd_modules/quantity.py
jax_dips/_jaxmd_modules/smap.py
jax_dips/_jaxmd_modules/space.py
jax_dips/_jaxmd_modules/util.py
jax_dips/data/__init__.py
jax_dips/data/data_management.py
jax_dips/data/data_stream.py
jax_dips/domain/__init__.py
jax_dips/domain/interpolate.py
jax_dips/domain/mesh.py
jax_dips/geometry/__init__.py
jax_dips/geometry/geometric_integrations.py
jax_dips/geometry/geometric_integrations_per_point.py
jax_dips/geometry/level_set.py
jax_dips/nn/__init__.py
jax_dips/nn/nn_solution_model.py
jax_dips/nn/nn_solution_trainer.py
jax_dips/solvers/__init__.py
jax_dips/solvers/optimizers.py
jax_dips/solvers/simulation_states.py
jax_dips/solvers/advection/__init__.py
jax_dips/solvers/advection/solver_advection.py
jax_dips/solvers/free_boundary/__init__.py
jax_dips/solvers/free_boundary/solver_poisson_advection.py
jax_dips/solvers/free_boundary/trainer_poisson_advection.py
jax_dips/solvers/poisson/__init__.py
jax_dips/solvers/poisson/discretization.py
jax_dips/solvers/poisson/nbm.py
jax_dips/solvers/poisson/trainer.py
jax_dips/utils/__init__.py
jax_dips/utils/chunking.py
jax_dips/utils/conversions.py
jax_dips/utils/inspect.py
jax_dips/utils/io.py
jax_dips/utils/visualization.py
tests/__init__.py
tests/test_advection.py
tests/test_amr.py
tests/test_geometric_integrations.py
tests/test_poisson.py
tests/test_reinitialization.py
tests/confs/__init__.py
tests/confs/experiment_configs.py