Index A | B | C | D | E | F | G | H | K | L | M | N | O | P | R | S | T | U A add_variance() (in module mellon.util) B batched_vmap() (in module mellon.util) C calculate_batch_elbo() (in module mellon.inference) calculate_elbo() (in module mellon.inference) calculate_gaussian_logpdf() (in module mellon.inference) compute_average_cell_count() (in module mellon.parameters) compute_conditional() (in module mellon.inference) (mellon.model.FunctionEstimator method) compute_conditional_explog() (in module mellon.inference) compute_conditional_times() (in module mellon.inference) compute_cov_func() (in module mellon.parameters) compute_d() (in module mellon.parameters) compute_d_factal() (in module mellon.parameters) compute_density_diffusion() (in module mellon.parameters) compute_density_gradient() (in module mellon.parameters) compute_dimensionality_loss_func() (in module mellon.inference) compute_dimensionality_transform() (in module mellon.inference) compute_distances() (in module mellon.parameters) compute_gp_type() (in module mellon.parameters) compute_initial_dimensionalities() (in module mellon.parameters) compute_initial_ones() (in module mellon.parameters) compute_initial_value() (in module mellon.parameters) compute_initial_zeros() (in module mellon.parameters) compute_L() (in module mellon.parameters) compute_landmarks() (in module mellon.parameters) compute_landmarks_rescale_time() (in module mellon.parameters) compute_log_density_x() (in module mellon.inference) compute_loss_func() (in module mellon.inference) compute_Lp() (in module mellon.parameters) compute_ls() (in module mellon.parameters) compute_mu() (in module mellon.parameters) compute_n_landmarks() (in module mellon.parameters) compute_nn_distances() (in module mellon.parameters) compute_nn_distances_within_time_points() (in module mellon.parameters) compute_parameter_cov_factor() (in module mellon.inference) compute_rank() (in module mellon.parameters) compute_time_derivatives() (in module mellon.parameters) compute_transform() (in module mellon.inference) covariance() (mellon.base_predictor.ExpPredictor method) (mellon.base_predictor.PredictorTime method) (mellon.Predictor method) D DensityEstimator (class in mellon.model) deserialize() (in module mellon.util) DimensionalityEstimator (class in mellon.model) distance() (in module mellon.util) distance_grad() (in module mellon.util) E ensure_2d() (in module mellon.util) expand_to_inactive() (in module mellon.util) Exponential (class in mellon.cov) ExpPredictor (class in mellon.base_predictor) ExpQuad (class in mellon.cov) F fit() (mellon.model.DensityEstimator method) (mellon.model.DimensionalityEstimator method) (mellon.model.FunctionEstimator method) (mellon.model.TimeSensitiveDensityEstimator method) fit_predict() (mellon.model.DensityEstimator method) (mellon.model.DimensionalityEstimator method) (mellon.model.FunctionEstimator method) (mellon.model.TimeSensitiveDensityEstimator method) from_dict() (mellon.Predictor class method) from_json() (mellon.Predictor class method) from_json_str() (mellon.Predictor class method) from_string() (mellon.util.GaussianProcessType static method) FULL (mellon.util.GaussianProcessType attribute) FULL_NYSTROEM (mellon.util.GaussianProcessType attribute) FunctionEstimator (class in mellon.model) G GaussianProcessType (class in mellon.util) generate_gaussian_sample() (in module mellon.inference) gradient() (mellon.base_predictor.PredictorTime method) (mellon.Predictor method) H hessian() (mellon.base_predictor.PredictorTime method) (mellon.Predictor method) hessian_log_determinant() (mellon.base_predictor.PredictorTime method) (mellon.Predictor method) K k() (mellon.cov.Exponential method) (mellon.cov.ExpQuad method) (mellon.cov.Linear method) (mellon.cov.Matern32 method) (mellon.cov.Matern52 method) (mellon.cov.RatQuad method) k_grad() (mellon.cov.Exponential method) (mellon.cov.ExpQuad method) (mellon.cov.Linear method) (mellon.cov.Matern32 method) (mellon.cov.Matern52 method) (mellon.cov.RatQuad method) L Linear (class in mellon.cov) local_dimensionality() (in module mellon.util) M make_multi_time_argument() (in module mellon.util) make_serializable() (in module mellon.util) Matern32 (class in mellon.cov) Matern52 (class in mellon.cov) mean() (mellon.base_predictor.ExpPredictor method) (mellon.base_predictor.PredictorTime method) (mellon.Predictor method) mean_covariance() (mellon.base_predictor.ExpPredictor method) (mellon.base_predictor.PredictorTime method) (mellon.Predictor method) mellon.cov module mellon.inference module mellon.model module mellon.parameters module mellon.util module minimize_adam() (in module mellon.inference) minimize_lbfgsb() (in module mellon.inference) mle() (in module mellon.util) module mellon.cov mellon.inference mellon.model mellon.parameters mellon.util N n_input_features (mellon.base_predictor.PredictorTime attribute) (mellon.Predictor attribute) n_obs (mellon.base_predictor.PredictorTime attribute) (mellon.Predictor attribute) O object_str() (in module mellon.util) P predict (mellon.model.DensityEstimator property) (mellon.model.DimensionalityEstimator property) (mellon.model.FunctionEstimator property) (mellon.model.TimeSensitiveDensityEstimator property) predict_density (mellon.model.DimensionalityEstimator property) Predictor (class in mellon) PredictorTime (class in mellon.base_predictor) prepare_inference() (mellon.model.DensityEstimator method) (mellon.model.DimensionalityEstimator method) (mellon.model.FunctionEstimator method) (mellon.model.TimeSensitiveDensityEstimator method) process_inference() (mellon.model.DensityEstimator method) (mellon.model.DimensionalityEstimator method) (mellon.model.TimeSensitiveDensityEstimator method) R RatQuad (class in mellon.cov) run_advi() (in module mellon.inference) run_inference() (mellon.model.DensityEstimator method) (mellon.model.DimensionalityEstimator method) (mellon.model.TimeSensitiveDensityEstimator method) S select_active_dims() (in module mellon.util) set_jax_config() (in module mellon.util) set_verbosity() (in module mellon.util) set_x() (mellon.model.DensityEstimator method) (mellon.model.DimensionalityEstimator method) (mellon.model.FunctionEstimator method) (mellon.model.TimeSensitiveDensityEstimator method) SPARSE_CHOLESKY (mellon.util.GaussianProcessType attribute) SPARSE_NYSTROEM (mellon.util.GaussianProcessType attribute) stabilize() (in module mellon.util) T test_rank() (in module mellon.util) time_derivative() (mellon.base_predictor.PredictorTime method) TimeSensitiveDensityEstimator (class in mellon.model) to_dict() (mellon.Predictor method) to_json() (mellon.Predictor method) U uncertainty() (mellon.base_predictor.ExpPredictor method) (mellon.base_predictor.PredictorTime method) (mellon.Predictor method)