LICENSE
README.md
pyproject.toml
src/omnibias/jax/__init__.py
src/omnibias/jax/_activations_piecewise.py
src/omnibias/jax/_activations_tempered.py
src/omnibias/jax/_fastpath.py
src/omnibias/jax/activations.py
src/omnibias/jax/activations_complex.py
src/omnibias/jax/bo_derivatives.py
src/omnibias/jax/confluence.py
src/omnibias/jax/confluent_bank.py
src/omnibias/jax/implicit.py
src/omnibias/jax/jet.py
src/omnibias/jax/jet_mv.py
src/omnibias/jax/laplacian.py
src/omnibias/jax/line_search.py
src/omnibias/jax/multipack.py
src/omnibias/jax/optim.py
src/omnibias/jax/optim_block_search.py
src/omnibias/jax/precision.py
src/omnibias/jax/py.typed
src/omnibias/jax/realization.py
src/omnibias/jax/refine.py
src/omnibias/jax/scan.py
src/omnibias/jax/scan_equivariant.py
src/omnibias/jax/weight_loss_jet.py
src/omnibias/jax/architectures/__init__.py
src/omnibias/jax/architectures/attention.py
src/omnibias/jax/architectures/ftc_net.py
src/omnibias/jax/architectures/hardbc.py
src/omnibias/jax/architectures/integral_kernel.py
src/omnibias/jax/architectures/jetkan.py
src/omnibias/jax/architectures/multiscale.py
src/omnibias/jax/architectures/pinn.py
src/omnibias/jax/architectures/piratenet.py
src/omnibias_jax.egg-info/PKG-INFO
src/omnibias_jax.egg-info/SOURCES.txt
src/omnibias_jax.egg-info/dependency_links.txt
src/omnibias_jax.egg-info/requires.txt
src/omnibias_jax.egg-info/top_level.txt
tests/test_audit_regressions.py
tests/test_block_exact_search.py
tests/test_bo_derivatives.py
tests/test_deep_laplacian.py
tests/test_ftc_net.py
tests/test_hardbc.py
tests/test_implicit.py
tests/test_integral_kernel.py
tests/test_jax_integral_primitives.py
tests/test_jax_limit.py
tests/test_jet.py
tests/test_jet_attention.py
tests/test_jet_mv.py
tests/test_jet_pinn.py
tests/test_laplacian_hessian.py
tests/test_line_search.py
tests/test_optim.py
tests/test_piecewise.py
tests/test_piratenet.py
tests/test_polylaplacian.py
tests/test_refine.py
tests/test_spectral_pinn.py
tests/test_tempered.py