File size: 1,563 Bytes
1cd8a52 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 | """Hackable surface for the multiscale polynomial memory research code."""
from __future__ import annotations
from .integration import IntegrationResult, integrate_system
from .models import AssocMemHiPPO, SalienceHiPPO, VanillaHiPPO, legendre_orthonormal_basis01
from .polynomials import (
EvalFn,
SystemParams,
fourier_val,
get_L_vector,
get_output_vector,
get_system_params,
legval,
legsval,
legval_jax,
legsval_jax,
make_m_mat,
)
from .synthetic import (
get_ou_sample,
oracle_mse_censored_task,
oracle_info_gain,
oracle_mse_rbf_periodic_censor,
oracle_info_gain_rbf,
make_copying_task,
make_censored_task,
make_induction_head_multioutput_s2s_task,
make_induction_head_seq_to_seq_task,
make_induction_head_task,
make_implicit_measure_task,
make_multiplexing_task,
whitesignal,
wray_and_green_output,
make_simple_repetition_task
)
from .utils import time_block, unvec, vec
__all__ = [
"AssocMemHiPPO",
"EvalFn",
"IntegrationResult",
"SystemParams",
"fourier_val",
"get_L_vector",
"get_output_vector",
"get_ou_sample",
"get_system_params",
"SalienceHiPPO",
"VanillaHiPPO",
"integrate_system",
"legendre_orthonormal_basis01",
"legval",
"legsval",
"make_induction_head_multioutput_s2s_task",
"make_induction_head_seq_to_seq_task",
"make_induction_head_task",
"make_implicit_measure_task",
"make_m_mat",
"make_multiplexing_task",
"time_block",
"unvec",
"vec",
]
|