Index

_

__init__() (nif.demo.TravelingWaveHighFreq method) _create_slots() (nif.optimizers.L4Adam method) _distributed_apply() (nif.optimizers.L4Adam method)

_momentum_add() (nif.optimizers.L4Adam method) _prepare_local() (nif.optimizers.L4Adam method) _resource_apply_dense() (nif.optimizers.L4Adam method) _resource_apply_sparse() (nif.optimizers.L4Adam method)

A

AdaBeliefOptimizer (class in nif.optimizers)

apply_gradients() (nif.optimizers.L4Adam method), [1]

B

BiasAddLayer (class in nif.layers)

build() (nif.layers.Dense method) (nif.model.NIF method), [1]

C

call() (nif.layers.BiasAddLayer method) (nif.layers.Dense method) (nif.layers.EinsumLayer method) (nif.layers.HessianLayer method) (nif.layers.HyperLinearForSIREN method), [1] (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), [1] (nif.layers.SIREN_ResNet method), [1] (nif.model.NIF method), [1] (nif.model.NIFMultiScale method) (nif.model.NIFMultiScaleLastLayerParameterized method), [1]

centralized_gradients_for_optimizer() (in module nif.optimizers) compute_output_shape() (nif.layers.Dense method) create_from_npz() (nif.data.TFRDataset method) CylinderFlow (class in nif.demo)

D

Dense (class in nif.layers)

E

EinsumLayer (class in nif.layers)

epoch (nif.optimizers.LBFGSOptimizer property)

F

function_factory() (in module nif.optimizers)

G

gen_dataset_from_batch_file() (nif.data.TFRDataset method) get_config() (nif.layers.BiasAddLayer method) (nif.layers.Dense method) (nif.layers.EinsumLayer method) (nif.layers.HyperLinearForSIREN method), [1] (nif.layers.JacRegLatentLayer method) (nif.layers.MLP_ResNet method) (nif.layers.MLP_SimpleShortCut method) (nif.layers.SIREN method), [1] (nif.optimizers.AdaBeliefOptimizer method) (nif.optimizers.L4Adam method), [1] (nif.optimizers.Lion method)

get_prunable_weights() (nif.layers.HyperLinearForSIREN method), [1] (nif.layers.MLP_ResNet method) (nif.layers.MLP_SimpleShortCut method) (nif.layers.SIREN method), [1] (nif.layers.SIREN_ResNet method), [1] get_tfr_meta_dataset() (nif.data.TFRDataset method)

H

HessianLayer (class in nif.layers)

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

J

JacobianLayer (class in nif.layers)

JacRegLatentLayer (class in nif.layers)

L

L4Adam (class in nif.optimizers) lbfgs_minimize() (in module nif.optimizers)

LBFGSOptimizer (class in nif.optimizers) Lion (class in nif.optimizers) loss (nif.optimizers.LBFGSOptimizer property)

M

minimize() (nif.optimizers.L4Adam method), [1] (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), [1] model_lr_to_w() (nif.model.NIF method), [1] (nif.model.NIFMultiScaleLastLayerParameterized method), [1] model_p_to_lr() (nif.model.NIF method), [1] (nif.model.NIFMultiScaleLastLayerParameterized method), [1]

model_p_to_w() (nif.model.NIF method), [1] model_x_to_phi() (nif.model.NIFMultiScaleLastLayerParameterized method), [1] model_x_to_u_given_w() (nif.model.NIF method), [1] (nif.model.NIFMultiScale method) (nif.model.NIFMultiScaleLastLayerParameterized method), [1] module nif.data nif.demo nif.layers nif.model nif.optimizers

N

NIF (class in nif.model) nif.data module nif.demo module nif.layers module

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

P

parameter (nif.data.PointWiseData property)

ParameterOutputL1ActReg (class in nif.layers) PointWiseData (class in nif.data)

S

save_config() (nif.model.NIF method), [1] set_weights() (nif.optimizers.AdaBeliefOptimizer method) SIREN (class in nif.layers)

SIREN_ResNet (class in nif.layers) standard_normalize() (nif.data.PointWiseData static method) (nif.demo.TravelingWaveHighFreq 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)

X

x (nif.data.PointWiseData property)