CHANGELOG.md
CITATION.cff
CONTRIBUTING.md
LICENSE
LIMITATIONS.md
MANIFEST.in
README.md
pyproject.toml
docs/api_overview.md
docs/determinism.md
docs/performance.md
docs/reproducibility.md
src/polystep/__init__.py
src/polystep/_compiled.py
src/polystep/_serialization.py
src/polystep/_step_blockwise.py
src/polystep/_step_core.py
src/polystep/_step_momentum.py
src/polystep/_step_monolithic.py
src/polystep/adaptive_subspace.py
src/polystep/api.py
src/polystep/ask_tell.py
src/polystep/blockwise.py
src/polystep/cma.py
src/polystep/cma_subspace.py
src/polystep/cost_nn.py
src/polystep/costs.py
src/polystep/dynamics.py
src/polystep/epsilon.py
src/polystep/factored_subspace.py
src/polystep/geometry.py
src/polystep/hybrid_subspace.py
src/polystep/optimizer.py
src/polystep/py.typed
src/polystep/quadratic_model.py
src/polystep/solver.py
src/polystep/subspace.py
src/polystep/transform.py
src/polystep.egg-info/PKG-INFO
src/polystep.egg-info/SOURCES.txt
src/polystep.egg-info/dependency_links.txt
src/polystep.egg-info/requires.txt
src/polystep.egg-info/top_level.txt
src/polystep/baselines/__init__.py
src/polystep/baselines/core.py
src/polystep/baselines/methods.py
src/polystep/benchmarks/__init__.py
src/polystep/benchmarks/threads.py
src/polystep/benchmarks/utils.py
src/polystep/benchmarks/rl/__init__.py
src/polystep/benchmarks/rl/cartpole.py
src/polystep/benchmarks/rl/gym_evaluator.py
src/polystep/benchmarks/rl/metrics.py
src/polystep/benchmarks/rl/policies.py
src/polystep/layers/__init__.py
src/polystep/layers/attention.py
src/polystep/layers/rnn.py
src/polystep/objectives/__init__.py
src/polystep/objectives/base.py
src/polystep/objectives/synthetic.py
src/polystep/projection/__init__.py
src/polystep/projection/sparse.py
src/polystep/solvers/__init__.py
src/polystep/solvers/_shared.py
src/polystep/solvers/base.py
src/polystep/solvers/greedy.py
src/polystep/solvers/kl_softmax.py
src/polystep/solvers/sinkhorn.py
src/polystep/solvers/softmax.py
src/polystep/solvers/tempered_softmax.py
tests/__init__.py
tests/conftest.py
tests/test_ablation_solvers.py
tests/test_adaptive_subspace.py
tests/test_amortization.py
tests/test_api.py
tests/test_ask_tell.py
tests/test_baseline_fairness.py
tests/test_baselines.py
tests/test_benchmarks.py
tests/test_biased_rotation.py
tests/test_blockwise.py
tests/test_candidate_autocast.py
tests/test_checkpoint_resume.py
tests/test_cma.py
tests/test_cma_subspace.py
tests/test_compiled.py
tests/test_cost_nn.py
tests/test_costs.py
tests/test_determinism.py
tests/test_dfo.py
tests/test_diagnostics.py
tests/test_dynamics.py
tests/test_end_to_end_learning.py
tests/test_epsilon.py
tests/test_evaluator_compile_forward.py
tests/test_factored_subspace.py
tests/test_fastpath_parity.py
tests/test_fused_projection.py
tests/test_geometry.py
tests/test_hybrid_subspace.py
tests/test_input_validation.py
tests/test_integration.py
tests/test_kl_softmax.py
tests/test_layers.py
tests/test_math_guards.py
tests/test_mixed_precision.py
tests/test_mnist.py
tests/test_no_test_set_leakage.py
tests/test_nondiff_data.py
tests/test_nondiff_models.py
tests/test_numerical_stress.py
tests/test_objectives.py
tests/test_optimizer.py
tests/test_per_sample.py
tests/test_probe_jitter.py
tests/test_quadratic_model.py
tests/test_regressions.py
tests/test_rl_benchmarks.py
tests/test_sequential_fast_path.py
tests/test_silent_failures.py
tests/test_sinkhorn.py
tests/test_sinkhorn_numerics.py
tests/test_site_vmap.py
tests/test_softmax.py
tests/test_softmax_integration.py
tests/test_sparse_delta_evaluator.py
tests/test_sparse_integration.py
tests/test_sparse_projection.py
tests/test_subspace_blockwise.py
tests/test_subspace_contract.py
tests/test_subspace_delta.py
tests/test_subspace_gain.py
tests/test_sweep_protocol.py
tests/test_timeseries_val_selection.py
tests/test_transform.py
tests/test_trust_region.py
tests/test_tuning_sweep.py
tests/test_vmap_layer_compatibility.py