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/conjugate.py
src/omnibias/jax/hierarchy.py
src/omnibias/jax/implicit.py
src/omnibias/jax/information.py
src/omnibias/jax/jet.py
src/omnibias/jax/jet_distill.py
src/omnibias/jax/jet_mv.py
src/omnibias/jax/laplacian.py
src/omnibias/jax/lindblad.py
src/omnibias/jax/line_search.py
src/omnibias/jax/moments.py
src/omnibias/jax/multipack.py
src/omnibias/jax/occupancy.py
src/omnibias/jax/optim.py
src/omnibias/jax/optim_block_search.py
src/omnibias/jax/optim_composed.py
src/omnibias/jax/optim_homotopy.py
src/omnibias/jax/optim_inverse.py
src/omnibias/jax/optim_kantorovich.py
src/omnibias/jax/optim_lqr.py
src/omnibias/jax/optim_maml.py
src/omnibias/jax/optim_mpc.py
src/omnibias/jax/optim_pid.py
src/omnibias/jax/optim_remainder.py
src/omnibias/jax/optim_sharp_loss.py
src/omnibias/jax/optim_sharpness.py
src/omnibias/jax/plant_pid.py
src/omnibias/jax/precision.py
src/omnibias/jax/probability.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/sequence.py
src/omnibias/jax/train_local.py
src/omnibias/jax/train_stack.py
src/omnibias/jax/transforms.py
src/omnibias/jax/weight_loss_jet.py
src/omnibias/jax/architectures/__init__.py
src/omnibias/jax/architectures/attention.py
src/omnibias/jax/architectures/collapse_net.py
src/omnibias/jax/architectures/frame_unet.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/jet_hopfield.py
src/omnibias/jax/architectures/jet_token.py
src/omnibias/jax/architectures/jetkan.py
src/omnibias/jax/architectures/ladder.py
src/omnibias/jax/architectures/multiscale.py
src/omnibias/jax/architectures/pack_moe.py
src/omnibias/jax/architectures/pinn.py
src/omnibias/jax/architectures/piratenet.py
src/omnibias/jax/architectures/riccati_flow.py
src/omnibias/jax/architectures/scannet.py
src/omnibias/jax/architectures/sliced_jet.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_collapse_net.py
tests/test_composed_curvature.py
tests/test_frame_unet.py
tests/test_ftc_net.py
tests/test_hardbc.py
tests/test_homotopy.py
tests/test_implicit.py
tests/test_information.py
tests/test_information_divergences.py
tests/test_integral_kernel.py
tests/test_inverse_design.py
tests/test_jax_integral_primitives.py
tests/test_jax_limit.py
tests/test_jet.py
tests/test_jet_attention.py
tests/test_jet_hopfield.py
tests/test_jet_mv.py
tests/test_jet_pinn.py
tests/test_jet_token.py
tests/test_kantorovich_accept.py
tests/test_laplacian_hessian.py
tests/test_line_search.py
tests/test_optim.py
tests/test_optim_maml.py
tests/test_optim_remainder.py
tests/test_pack_moe.py
tests/test_piecewise.py
tests/test_piratenet.py
tests/test_polylaplacian.py
tests/test_probability.py
tests/test_refine.py
tests/test_riccati_flow.py
tests/test_sequence.py
tests/test_sharp_loss.py
tests/test_sharpness_schedule.py
tests/test_spectral_pinn.py
tests/test_tempered.py
tests/test_train_local.py
tests/test_transforms.py