.gitignore
.mlc.external.json
.mlc.relative.json
.pre-commit-config.yaml
.readthedocs.yaml
CCN-letterFoot.png
CITATION.cff
CODE_OF_CONDUCT.md
CONTRIBUTING.md
LICENSE
README.md
codecov.yml
pyproject.toml
tox.ini
.github/CODEOWNERS
.github/ISSUE_TEMPLATE/bug_report.md
.github/ISSUE_TEMPLATE/feature_request.md
.github/workflows/ci.yml
.github/workflows/connect.yml
.github/workflows/deploy-pure-python.yml
docs/Makefile
docs/api_reference.rst
docs/citation.md
docs/conf.py
docs/getting_help.md
docs/index.md
docs/installation.md
docs/make.bat
docs/quickstart.md
docs/references.bib
docs/_templates/layout.html
docs/_templates/autosummary/class.rst
docs/assets/CCN-logo-wText.png
docs/assets/NeMoS_Icon_CMYK_Full.svg
docs/assets/NeMoS_Icon_CMYK_White.svg
docs/assets/NeMoS_Logo_CMYK_Full.svg
docs/assets/NeMoS_Logo_CMYK_White.svg
docs/assets/NeMoS_favicon.ico
docs/assets/allen_data.png
docs/assets/allen_data_annotated-epochs.svg
docs/assets/allen_data_annotated-response.svg
docs/assets/allen_data_annotated-stimulus.svg
docs/assets/allen_data_annotated-units.svg
docs/assets/allen_data_annotated.gif
docs/assets/classes_nemos.png
docs/assets/classes_nemos.svg
docs/assets/convolve_batching_scheme.svg
docs/assets/glm_features_scheme.svg
docs/assets/glm_population_scheme.svg
docs/assets/glm_scheme.svg
docs/assets/head_dir_tuning.jpg
docs/assets/kfold.svg
docs/assets/lnp_model.svg
docs/assets/logo_flatiron_white.svg
docs/assets/nemos_sklearn.svg
docs/assets/pipeline.svg
docs/assets/poisson_model_calc_stepsize.pdf
docs/assets/thumbnail_save_load.svg
docs/assets/stylesheets/custom.css
docs/background/README.md
docs/background/plot_00_conceptual_intro.md
docs/background/plot_03_1D_convolution.md
docs/background/basis/README.md
docs/background/basis/ndimensional_foureir_basis.md
docs/background/basis/plot_01_1D_basis_function.md
docs/background/basis/plot_02_ND_basis_function.md
docs/developers_notes/01-base_class.md
docs/developers_notes/02-base_regressor.md
docs/developers_notes/03-glm.md
docs/developers_notes/04-basis_module.md
docs/developers_notes/05-observation_models.md
docs/developers_notes/06-regularizer.md
docs/developers_notes/07-solvers.md
docs/developers_notes/README.md
docs/how_to_guide/README.md
docs/how_to_guide/categorical_predictors.md
docs/how_to_guide/convolve_large_arrays.md
docs/how_to_guide/custom_predictors.md
docs/how_to_guide/define_a_custom_basis.md
docs/how_to_guide/handling_composite_bases.md
docs/how_to_guide/plot_02_glm_demo.md
docs/how_to_guide/plot_03_population_glm.md
docs/how_to_guide/plot_04_batch_glm.md
docs/how_to_guide/plot_05_transformer_basis.md
docs/how_to_guide/plot_06_sklearn_pipeline_cv_demo.md
docs/how_to_guide/plot_07_glm_pytree.md
docs/how_to_guide/raw_history_feature.md
docs/how_to_guide/save_and_load.md
docs/javascripts/katex.js
docs/scripts/basis_figs.py
docs/scripts/glm_predictors.py
docs/tutorials/README.md
docs/tutorials/plot_01_current_injection.md
docs/tutorials/plot_02_head_direction.md
docs/tutorials/plot_03_grid_cells.md
docs/tutorials/plot_04_v1_cells.md
docs/tutorials/plot_05_place_cells.md
docs/tutorials/plot_06_calcium_imaging.md
scripts/check_markdown_links.sh
scripts/check_parameter_naming.py
scripts/prevent_absolute_links_to_docs.sh
src/nemos/__init__.py
src/nemos/_observation_model_builder.py
src/nemos/_regularizer_builder.py
src/nemos/base_class.py
src/nemos/base_regressor.py
src/nemos/convolve.py
src/nemos/exceptions.py
src/nemos/identifiability_constraints.py
src/nemos/observation_models.py
src/nemos/proximal_operator.py
src/nemos/pytrees.py
src/nemos/regularizer.py
src/nemos/simulation.py
src/nemos/tree_utils.py
src/nemos/type_casting.py
src/nemos/typing.py
src/nemos/utils.py
src/nemos/validation.py
src/nemos.egg-info/PKG-INFO
src/nemos.egg-info/SOURCES.txt
src/nemos.egg-info/dependency_links.txt
src/nemos.egg-info/requires.txt
src/nemos.egg-info/top_level.txt
src/nemos/_documentation_utils/__init__.py
src/nemos/_documentation_utils/_myst_nb_glue.py
src/nemos/_documentation_utils/plotting.py
src/nemos/_inspect_utils/__init__.py
src/nemos/_inspect_utils/inspect_utils.py
src/nemos/basis/__init__.py
src/nemos/basis/_basis.py
src/nemos/basis/_basis_mixin.py
src/nemos/basis/_check_basis.py
src/nemos/basis/_composition_utils.py
src/nemos/basis/_custom_basis.py
src/nemos/basis/_decaying_exponential.py
src/nemos/basis/_fourier_basis.py
src/nemos/basis/_identity.py
src/nemos/basis/_raised_cosine_basis.py
src/nemos/basis/_spline_basis.py
src/nemos/basis/_transformer_basis.py
src/nemos/basis/basis.py
src/nemos/fetch/__init__.py
src/nemos/fetch/fetch_data.py
src/nemos/glm/__init__.py
src/nemos/glm/glm.py
src/nemos/glm/initialize_parameters.py
src/nemos/glm/inverse_link_function_utils.py
src/nemos/glm_hmm/__init__.py
src/nemos/glm_hmm/expectation_maximization.py
src/nemos/io/__init__.py
src/nemos/io/io.py
src/nemos/solvers/__init__.py
src/nemos/solvers/_abstract_solver.py
src/nemos/solvers/_compute_defaults.py
src/nemos/solvers/_jaxopt_solvers.py
src/nemos/solvers/_optax_optimistix_solvers.py
src/nemos/solvers/_optimistix_solvers.py
src/nemos/solvers/_solver_adapter.py
src/nemos/solvers/_solver_doc_helper.py
src/nemos/solvers/_solver_registry.py
src/nemos/solvers/_svrg.py
src/nemos/solvers/_svrg_defaults.py
src/nemos/styles/__init__.py
src/nemos/styles/nemos.mplstyle
src/nemos/third_party/README.md
src/nemos/third_party/jaxopt/.gitignore
src/nemos/third_party/jaxopt/.pre-commit-config.yaml
src/nemos/third_party/jaxopt/CONTRIBUTING.md
src/nemos/third_party/jaxopt/LICENSE
src/nemos/third_party/jaxopt/README.md
src/nemos/third_party/jaxopt/pylintrc
src/nemos/third_party/jaxopt/requirements.txt
src/nemos/third_party/jaxopt/requirements_test.txt
src/nemos/third_party/jaxopt/setup.py
src/nemos/third_party/jaxopt/jaxopt/__init__.py
src/nemos/third_party/jaxopt/jaxopt/base.py
src/nemos/third_party/jaxopt/jaxopt/cond.py
src/nemos/third_party/jaxopt/jaxopt/implicit_diff.py
src/nemos/third_party/jaxopt/jaxopt/isotonic.py
src/nemos/third_party/jaxopt/jaxopt/linear_solve.py
src/nemos/third_party/jaxopt/jaxopt/loop.py
src/nemos/third_party/jaxopt/jaxopt/loss.py
src/nemos/third_party/jaxopt/jaxopt/objective.py
src/nemos/third_party/jaxopt/jaxopt/perturbations.py
src/nemos/third_party/jaxopt/jaxopt/projection.py
src/nemos/third_party/jaxopt/jaxopt/prox.py
src/nemos/third_party/jaxopt/jaxopt/tree_util.py
src/nemos/third_party/jaxopt/jaxopt/version.py
src/nemos/third_party/jaxopt/jaxopt/_src/__init__.py
src/nemos/third_party/jaxopt/jaxopt/_src/anderson.py
src/nemos/third_party/jaxopt/jaxopt/_src/anderson_wrapper.py
src/nemos/third_party/jaxopt/jaxopt/_src/armijo_sgd.py
src/nemos/third_party/jaxopt/jaxopt/_src/backtracking_linesearch.py
src/nemos/third_party/jaxopt/jaxopt/_src/base.py
src/nemos/third_party/jaxopt/jaxopt/_src/bfgs.py
src/nemos/third_party/jaxopt/jaxopt/_src/bisection.py
src/nemos/third_party/jaxopt/jaxopt/_src/block_cd.py
src/nemos/third_party/jaxopt/jaxopt/_src/broyden.py
src/nemos/third_party/jaxopt/jaxopt/_src/cd_qp.py
src/nemos/third_party/jaxopt/jaxopt/_src/cond.py
src/nemos/third_party/jaxopt/jaxopt/_src/cvxpy_wrapper.py
src/nemos/third_party/jaxopt/jaxopt/_src/eq_qp.py
src/nemos/third_party/jaxopt/jaxopt/_src/fixed_point_iteration.py
src/nemos/third_party/jaxopt/jaxopt/_src/gauss_newton.py
src/nemos/third_party/jaxopt/jaxopt/_src/gradient_descent.py
src/nemos/third_party/jaxopt/jaxopt/_src/hager_zhang_linesearch.py
src/nemos/third_party/jaxopt/jaxopt/_src/implicit_diff.py
src/nemos/third_party/jaxopt/jaxopt/_src/isotonic.py
src/nemos/third_party/jaxopt/jaxopt/_src/iterative_refinement.py
src/nemos/third_party/jaxopt/jaxopt/_src/lbfgs.py
src/nemos/third_party/jaxopt/jaxopt/_src/lbfgsb.py
src/nemos/third_party/jaxopt/jaxopt/_src/levenberg_marquardt.py
src/nemos/third_party/jaxopt/jaxopt/_src/linear_operator.py
src/nemos/third_party/jaxopt/jaxopt/_src/linear_solve.py
src/nemos/third_party/jaxopt/jaxopt/_src/linesearch_util.py
src/nemos/third_party/jaxopt/jaxopt/_src/loop.py
src/nemos/third_party/jaxopt/jaxopt/_src/loss.py
src/nemos/third_party/jaxopt/jaxopt/_src/mirror_descent.py
src/nemos/third_party/jaxopt/jaxopt/_src/nonlinear_cg.py
src/nemos/third_party/jaxopt/jaxopt/_src/objective.py
src/nemos/third_party/jaxopt/jaxopt/_src/optax_wrapper.py
src/nemos/third_party/jaxopt/jaxopt/_src/osqp.py
src/nemos/third_party/jaxopt/jaxopt/_src/perturbations.py
src/nemos/third_party/jaxopt/jaxopt/_src/polyak_sgd.py
src/nemos/third_party/jaxopt/jaxopt/_src/projected_gradient.py
src/nemos/third_party/jaxopt/jaxopt/_src/projection.py
src/nemos/third_party/jaxopt/jaxopt/_src/prox.py
src/nemos/third_party/jaxopt/jaxopt/_src/proximal_gradient.py
src/nemos/third_party/jaxopt/jaxopt/_src/scipy_wrappers.py
src/nemos/third_party/jaxopt/jaxopt/_src/test_util.py
src/nemos/third_party/jaxopt/jaxopt/_src/tree_util.py
src/nemos/third_party/jaxopt/jaxopt/_src/zoom_linesearch.py
src/nemos/third_party/jaxopt/tests/anderson_test.py
src/nemos/third_party/jaxopt/tests/anderson_wrapper_test.py
src/nemos/third_party/jaxopt/tests/armijo_sgd_test.py
src/nemos/third_party/jaxopt/tests/backtracking_linesearch_test.py
src/nemos/third_party/jaxopt/tests/base_test.py
src/nemos/third_party/jaxopt/tests/bfgs_test.py
src/nemos/third_party/jaxopt/tests/bisection_test.py
src/nemos/third_party/jaxopt/tests/block_cd_test.py
src/nemos/third_party/jaxopt/tests/broyden_test.py
src/nemos/third_party/jaxopt/tests/cd_qp_test.py
src/nemos/third_party/jaxopt/tests/common_test.py
src/nemos/third_party/jaxopt/tests/cond_test.py
src/nemos/third_party/jaxopt/tests/cvxpy_wrapper_test.py
src/nemos/third_party/jaxopt/tests/eq_qp_test.py
src/nemos/third_party/jaxopt/tests/fixed_point_iteration_test.py
src/nemos/third_party/jaxopt/tests/gauss_newton_test.py
src/nemos/third_party/jaxopt/tests/gradient_descent_test.py
src/nemos/third_party/jaxopt/tests/hager_zhang_linesearch_test.py
src/nemos/third_party/jaxopt/tests/implicit_diff_test.py
src/nemos/third_party/jaxopt/tests/import_test.py
src/nemos/third_party/jaxopt/tests/isotonic_test.py
src/nemos/third_party/jaxopt/tests/iterative_refinement_test.py
src/nemos/third_party/jaxopt/tests/lbfgs_test.py
src/nemos/third_party/jaxopt/tests/lbfgsb_test.py
src/nemos/third_party/jaxopt/tests/levenberg_marquardt_test.py
src/nemos/third_party/jaxopt/tests/linear_operator_test.py
src/nemos/third_party/jaxopt/tests/linear_solve_test.py
src/nemos/third_party/jaxopt/tests/linesearch_common_test.py
src/nemos/third_party/jaxopt/tests/loop_test.py
src/nemos/third_party/jaxopt/tests/loss_test.py
src/nemos/third_party/jaxopt/tests/mirror_descent_test.py
src/nemos/third_party/jaxopt/tests/nonlinear_cg_test.py
src/nemos/third_party/jaxopt/tests/optax_wrapper_test.py
src/nemos/third_party/jaxopt/tests/osqp_test.py
src/nemos/third_party/jaxopt/tests/perturbations_test.py
src/nemos/third_party/jaxopt/tests/polyak_sgd_test.py
src/nemos/third_party/jaxopt/tests/projected_gradient_test.py
src/nemos/third_party/jaxopt/tests/projection_test.py
src/nemos/third_party/jaxopt/tests/prox_test.py
src/nemos/third_party/jaxopt/tests/proximal_gradient_test.py
src/nemos/third_party/jaxopt/tests/scipy_wrappers_test.py
src/nemos/third_party/jaxopt/tests/tree_util_test.py
src/nemos/third_party/jaxopt/tests/zoom_linesearch_test.py
tests/conftest.py
tests/simulate_coupled_neurons_params.json
tests/test_base_class.py
tests/test_basis.py
tests/test_composition_utils.py
tests/test_convergence.py
tests/test_convolution.py
tests/test_custom_basis.py
tests/test_glm.py
tests/test_glm_hmm.py
tests/test_glm_initialization.py
tests/test_identifiability_constraints.py
tests/test_inspect_utils.py
tests/test_myst_nb_glue_utils.py
tests/test_observation_models.py
tests/test_optimistix_dtype.py
tests/test_pipeline.py
tests/test_proximal_operator.py
tests/test_pytrees.py
tests/test_regularizer.py
tests/test_simulation.py
tests/test_solvers.py
tests/test_svrg_defaults.py
tests/test_transformer_basis.py
tests/test_tree_utils.py
tests/test_type_casting.py
tests/test_utils.py
tests/test_validation.py