Index _ | A | B | C | D | F | G | I | L | M | N | P | R | S | T | V _ __init__() (trak.gradient_computers.AbstractGradientComputer method) (trak.gradient_computers.FunctionalGradientComputer method) (trak.gradient_computers.IterativeGradientComputer method) (trak.modelout_functions.AbstractModelOutput method) (trak.modelout_functions.CLIPModelOutput method) (trak.modelout_functions.ImageClassificationModelOutput method) (trak.modelout_functions.IterativeImageClassificationModelOutput method) (trak.modelout_functions.TextClassificationModelOutput method) (trak.projectors.AbstractProjector method) (trak.projectors.BasicProjector method) (trak.projectors.BasicSingleBlockProjector method) (trak.projectors.ChunkedCudaProjector method) (trak.projectors.CudaProjector method) (trak.projectors.NoOpProjector method) (trak.savers.AbstractSaver method) (trak.savers.MmapSaver method) (trak.score_computers.AbstractScoreComputer method) (trak.score_computers.BasicScoreComputer method) (trak.traker.TRAKer method) A AbstractGradientComputer (class in trak.gradient_computers) AbstractModelOutput (class in trak.modelout_functions) AbstractProjector (class in trak.projectors) AbstractSaver (class in trak.savers) AbstractScoreComputer (class in trak.score_computers) allocate_input() (trak.projectors.ChunkedCudaProjector method) B BasicProjector (class in trak.projectors) BasicScoreComputer (class in trak.score_computers) BasicSingleBlockProjector (class in trak.projectors) BasicSingleBlockScoreComputer (class in trak.score_computers) C ChunkedCudaProjector (class in trak.projectors) CLIPModelOutput (class in trak.modelout_functions) compute_loss_grad() (trak.gradient_computers.AbstractGradientComputer method) (trak.gradient_computers.FunctionalGradientComputer method) (trak.gradient_computers.IterativeGradientComputer method) compute_per_sample_grad() (trak.gradient_computers.AbstractGradientComputer method) (trak.gradient_computers.FunctionalGradientComputer method) (trak.gradient_computers.IterativeGradientComputer method) CudaProjector (class in trak.projectors) D del_grads() (trak.savers.AbstractSaver method) (trak.savers.MmapSaver method) F featurize() (trak.traker.TRAKer method) finalize_features() (trak.traker.TRAKer method) finalize_scores() (trak.traker.TRAKer method) free_memory() (trak.projectors.AbstractProjector method) (trak.projectors.BasicProjector method) (trak.projectors.BasicSingleBlockProjector method) (trak.projectors.ChunkedCudaProjector method) (trak.projectors.CudaProjector method) (trak.projectors.NoOpProjector method) FunctionalGradientComputer (class in trak.gradient_computers) G generate_sketch_matrix() (trak.projectors.BasicProjector method) (trak.projectors.BasicSingleBlockProjector method) get_embeddings() (trak.modelout_functions.CLIPModelOutput static method) get_free_memory() (in module trak.utils) get_generator_states() (trak.projectors.BasicProjector method) get_matrix_mult() (in module trak.utils) get_matrix_mult_blockwise() (in module trak.utils) get_matrix_mult_standard() (in module trak.utils) get_num_params() (in module trak.utils) get_out_to_loss_grad() (trak.modelout_functions.AbstractModelOutput method) (trak.modelout_functions.CLIPModelOutput method) (trak.modelout_functions.ImageClassificationModelOutput method) (trak.modelout_functions.IterativeImageClassificationModelOutput method) (trak.modelout_functions.TextClassificationModelOutput method) get_output() (trak.modelout_functions.AbstractModelOutput method) (trak.modelout_functions.CLIPModelOutput static method) (trak.modelout_functions.ImageClassificationModelOutput static method) (trak.modelout_functions.IterativeImageClassificationModelOutput static method) (trak.modelout_functions.TextClassificationModelOutput static method) get_output_memory() (in module trak.utils) get_parameter_chunk_sizes() (in module trak.utils) get_scores() (trak.score_computers.AbstractScoreComputer method) (trak.score_computers.BasicScoreComputer method) (trak.score_computers.BasicSingleBlockScoreComputer method) get_x_xtx_inv() (trak.score_computers.AbstractScoreComputer method) (trak.score_computers.BasicScoreComputer method) (trak.score_computers.BasicSingleBlockScoreComputer method) get_xtx() (trak.score_computers.AbstractScoreComputer method) (trak.score_computers.BasicScoreComputer method) (trak.score_computers.BasicSingleBlockScoreComputer method) I image_embeddings (trak.modelout_functions.CLIPModelOutput attribute) ImageClassificationModelOutput (class in trak.modelout_functions) init_experiment() (trak.savers.AbstractSaver method) (trak.savers.MmapSaver method) init_projector() (trak.traker.TRAKer method) init_store() (trak.savers.AbstractSaver method) (trak.savers.MmapSaver method) is_not_buffer() (in module trak.utils) IterativeGradientComputer (class in trak.gradient_computers) IterativeImageClassificationModelOutput (class in trak.modelout_functions) L load_checkpoint() (trak.traker.TRAKer method) load_current_store() (trak.savers.AbstractSaver method) (trak.savers.MmapSaver method) load_model_params() (trak.gradient_computers.AbstractGradientComputer method) (trak.gradient_computers.FunctionalGradientComputer method) (trak.gradient_computers.IterativeGradientComputer method) M MmapSaver (class in trak.savers) ModelIDException module trak trak.gradient_computers trak.modelout_functions trak.projectors trak.savers trak.score_computers trak.traker trak.utils N NoOpProjector (class in trak.projectors) normal (trak.projectors.ProjectionType attribute) num_computed_embeddings (trak.modelout_functions.CLIPModelOutput attribute) P parameters_to_vector() (in module trak.utils) project() (trak.projectors.AbstractProjector method) (trak.projectors.BasicProjector method) (trak.projectors.BasicSingleBlockProjector method) (trak.projectors.ChunkedCudaProjector method) (trak.projectors.CudaProjector method) (trak.projectors.NoOpProjector method) ProjectionType (class in trak.projectors) R rademacher (trak.projectors.ProjectionType attribute) register_model_id() (trak.savers.AbstractSaver method) (trak.savers.MmapSaver method) S save_scores() (trak.savers.AbstractSaver method) (trak.savers.MmapSaver method) score() (trak.traker.TRAKer method) serialize_current_model_id_metadata() (trak.savers.AbstractSaver method) (trak.savers.MmapSaver method) sim_batch_size (trak.modelout_functions.CLIPModelOutput attribute) start_scoring_checkpoint() (trak.traker.TRAKer method) T test_install() (in module trak.utils) text_embeddings (trak.modelout_functions.CLIPModelOutput attribute) TextClassificationModelOutput (class in trak.modelout_functions) trak module trak.gradient_computers module trak.modelout_functions module trak.projectors module trak.savers module trak.score_computers module trak.traker module trak.utils module TRAKer (class in trak.traker) V vectorize() (in module trak.utils)