Index

_

__init__() (nif.demo.TravelingWaveHighFreq method)

A

AdaBeliefOptimizer (class in nif.optimizers)

assign_tensors() (in module nif.optimizers)

B

BiasAddLayer (class in nif.layers)

build() (nif.model.NIF method)

C

call() (nif.model.NIF method) centralized_gradients_for_optimizer() (in module nif.optimizers) compute_Dtype (nif.model.NIF property) compute_number_of_weightbias_by_its_position_for_shapenet() (in module nif.layers) compute_output_and_augment_grad() (in module nif.layers)

compute_output_and_grad() (in module nif.layers) compute_output_and_grad_and_hessian() (in module nif.layers) create_from_npz() (nif.data.TFRDataset method) create_value_and_gradient_function() (in module nif.optimizers) CylinderFlow (class in nif.demo)

E

EinsumLayer (class in nif.layers)

epoch (nif.optimizers.LBFGSOptimizer property)

F

forward() (nif.layers.BiasAddLayer method) (nif.layers.EinsumLayer method) (nif.layers.HessianLayer method) (nif.layers.HyperLinearForSIREN method) (nif.layers.JacobianLayer method) (nif.layers.JacRegLatentLayer method) (nif.layers.MLP_ResNet method) (nif.layers.MLP_SimpleShortCut method) (nif.layers.ParameterOutputL1ActReg method) (nif.layers.SIREN method) (nif.layers.SIREN_ResNet method) (nif.model.NIF method)

function_factory() (in module nif.optimizers)

G

gen_dataset_from_batch_file() (nif.data.TFRDataset method) gen_hypernetwork_weights_bias_for_siren_shapenet() (in module nif.layers) get_centralized_gradients() (in module nif.optimizers) get_config() (nif.layers.BiasAddLayer method) (nif.layers.EinsumLayer method) (nif.layers.HyperLinearForSIREN method) (nif.layers.JacobianLayer method) (nif.layers.JacRegLatentLayer method) (nif.layers.MLP_ResNet method) (nif.layers.MLP_SimpleShortCut method) (nif.layers.ParameterOutputL1ActReg method) (nif.layers.SIREN method) (nif.layers.SIREN_ResNet method)

get_prunable_weights() (nif.layers.HyperLinearForSIREN method) (nif.layers.MLP_ResNet method) (nif.layers.MLP_SimpleShortCut method) (nif.layers.SIREN method) (nif.layers.SIREN_ResNet method) get_tfr_meta_dataset() (nif.data.TFRDataset method)

H

HessianLayer (class in nif.layers)

history (nif.optimizers.TFPLBFGS property) HyperLinearForSIREN (class in nif.layers)

I

inverse_normalize() (nif.data.PointWiseData static method)

J

JacobianLayer (class in nif.layers)

JacRegLatentLayer (class in nif.layers)

L

L4Adam (class in nif.optimizers) l_min (nif.optimizers.L4Adam property) lbfgs_minimize() (in module nif.optimizers) LBFGSOptimizer (class in nif.optimizers)

Lion (class in nif.optimizers) load_state_dict() (nif.optimizers.ReferenceAdam method) (nif.optimizers.ReferenceLBFGS method) loss (nif.optimizers.LBFGSOptimizer property)

M

mask (nif.pruning.PolynomialMagnitudePruning property) minimize() (nif.optimizers.LBFGSOptimizer method) (nif.optimizers.TFPLBFGS method) minmax_normalize() (nif.data.PointWiseData static method) (nif.demo.TravelingWaveHighFreq method) MLP_ResNet (class in nif.layers) MLP_SimpleShortCut (class in nif.layers)

model() (nif.model.NIF method) model_lr_to_w() (nif.model.NIF method) (nif.model.NIFMultiScaleLastLayerParameterized method) model_p_to_lr() (nif.model.NIF method) (nif.model.NIFMultiScaleLastLayerParameterized method) model_p_to_w() (nif.model.NIF method) model_x_to_phi() (nif.model.NIFMultiScaleLastLayerParameterized method) model_x_to_u_given_w() (nif.model.NIF method)

N

NIF (class in nif.model) nif.layers.Dense (built-in class)

NIFMultiScale (class in nif.model) NIFMultiScaleLastLayerParameterized (class in nif.model)

O

on_epoch_end() (nif.pruning.PolynomialMagnitudePruning method)

P

pack_tensors() (in module nif.optimizers) parameter (nif.data.PointWiseData property) ParameterOutputL1ActReg (class in nif.layers)

PointWiseData (class in nif.data) PolynomialMagnitudePruning (class in nif.pruning) predict() (nif.model.NIF method)

R

ReferenceAdam (class in nif.optimizers) ReferenceLBFGS (class in nif.optimizers) refresh_count (nif.pruning.PolynomialMagnitudePruning property) regularization_loss() (nif.layers.HyperLinearForSIREN method) (nif.layers.JacRegLatentLayer method) (nif.layers.MLP_ResNet method) (nif.layers.MLP_SimpleShortCut method) (nif.layers.ParameterOutputL1ActReg method) (nif.layers.SIREN method) (nif.model.NIF method)

S

save_config() (nif.model.NIF method) SIREN (class in nif.layers) SIREN_ResNet (class in nif.layers) standard_normalize() (nif.data.PointWiseData static method) (nif.demo.TravelingWaveHighFreq method) state_dict() (nif.optimizers.ReferenceAdam method) (nif.optimizers.ReferenceLBFGS method)

step (nif.pruning.PolynomialMagnitudePruning property) step() (nif.optimizers.AdaBeliefOptimizer method) (nif.optimizers.L4Adam method) (nif.optimizers.Lion method) (nif.optimizers.ReferenceAdam method) (nif.optimizers.ReferenceLBFGS method) strip() (nif.pruning.PolynomialMagnitudePruning method)

T

TFPLBFGS (class in nif.optimizers) TFRDataset (class in nif.data)

TravelingWave (class in nif.demo) TravelingWaveHighFreq (class in nif.demo)

U

u (nif.data.PointWiseData property)

unpack_tensors() (in module nif.optimizers)

V

variable_Dtype (nif.model.NIF property)

X

x (nif.data.PointWiseData property)