LICENSE
README.md
pyproject.toml
src/polystep/__init__.py
src/polystep/_compiled.py
src/polystep/_step_blockwise.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/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/benchmarks/__init__.py
src/polystep/benchmarks/baselines.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/_prelude.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/test_ablation_solvers.py
tests/test_adaptive_mnist.py
tests/test_adaptive_subspace.py
tests/test_api.py
tests/test_ask_tell.py
tests/test_baseline_fairness.py
tests/test_benchmarks.py
tests/test_biased_rotation.py
tests/test_blockwise.py
tests/test_bug_regressions.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_dfo_features.py
tests/test_dynamics.py
tests/test_fullspace_chunk_bound.py
tests/test_fused_projection_rebuild.py
tests/test_geometry.py
tests/test_hardening.py
tests/test_hybrid_mnist.py
tests/test_hybrid_subspace.py
tests/test_integration.py
tests/test_kl_softmax.py
tests/test_layers.py
tests/test_mixed_precision.py
tests/test_mixed_precision_step.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_optimizer.py
tests/test_quadratic_model.py
tests/test_rl_benchmarks.py
tests/test_sinkhorn.py
tests/test_sinkhorn_correctness.py
tests/test_softmax.py
tests/test_softmax_integration.py
tests/test_solver_numerics.py
tests/test_sparse_integration.py
tests/test_sparse_projection.py
tests/test_subspace.py
tests/test_subspace_blockwise.py
tests/test_transform.py
tests/test_trust_region.py
tests/test_turbo.py
tests/test_vmap_layer_compatibility.py