Index

_

__call__() (objax.ForceArgs method) (objax.Function method) (objax.Grad method) (objax.GradValues method) (objax.jaxboard.Summary method) (objax.Jit method) (objax.Module method) (objax.nn.BatchNorm method) (objax.nn.BatchNorm0D method) (objax.nn.BatchNorm1D method) (objax.nn.BatchNorm2D method) (objax.nn.Conv2D method) (objax.nn.ConvTranspose2D method) (objax.nn.Dropout method) (objax.nn.ExponentialMovingAverage method) (objax.nn.Linear method) (objax.nn.MovingAverage method) (objax.nn.Sequential method) (objax.nn.SyncedBatchNorm method) (objax.nn.SyncedBatchNorm0D method) (objax.nn.SyncedBatchNorm1D method) (objax.nn.SyncedBatchNorm2D method) (objax.optimizer.Adam method) (objax.optimizer.ExponentialMovingAverage method) (objax.optimizer.ExponentialMovingAverageModule method) (objax.optimizer.LARS method) (objax.optimizer.Momentum method) (objax.optimizer.SGD method) (objax.Parallel method) (objax.privacy.dpsgd.PrivateGradValues method) (objax.random.Generator method) (objax.util.Objax2Tf method) (objax.util.Renamer method) (objax.Vectorize method) (objax.zoo.rnn.RNN method) (objax.zoo.wide_resnet.WRNBlock method) __init__() (objax.BaseState method) (objax.BaseVar method) (objax.ForceArgs method) (objax.Function method) (objax.Grad method) (objax.GradValues method) (objax.Hessian method) (objax.io.Checkpoint method) (objax.Jacobian method) (objax.jaxboard.SummaryWriter method) (objax.Jit method) (objax.nn.BatchNorm method) (objax.nn.BatchNorm0D method) (objax.nn.BatchNorm1D method) (objax.nn.BatchNorm2D method) (objax.nn.Conv2D method) (objax.nn.ConvTranspose2D method) (objax.nn.Dropout method) (objax.nn.ExponentialMovingAverage method) (objax.nn.Linear method) (objax.nn.MovingAverage method) (objax.nn.Sequential method) (objax.nn.SyncedBatchNorm method) (objax.nn.SyncedBatchNorm0D method) (objax.nn.SyncedBatchNorm1D method) (objax.nn.SyncedBatchNorm2D method) (objax.optimizer.Adam method) (objax.optimizer.ExponentialMovingAverage method) (objax.optimizer.ExponentialMovingAverageModule method) (objax.optimizer.LARS method) (objax.optimizer.Momentum method) (objax.optimizer.SGD method) (objax.Parallel method) (objax.privacy.dpsgd.PrivateGradValues method) (objax.random.Generator method) (objax.RandomState method) (objax.StateVar method) (objax.TrainRef method) (objax.TrainVar method) (objax.util.EasyDict method) (objax.util.Objax2Tf method) (objax.util.Renamer method) (objax.VarCollection method) (objax.Vectorize method) (objax.zoo.convnet.ConvNet method) (objax.zoo.dnnet.DNNet method) (objax.zoo.resnet_v2.ResNet101 method) (objax.zoo.resnet_v2.ResNet152 method) (objax.zoo.resnet_v2.ResNet18 method) (objax.zoo.resnet_v2.ResNet200 method) (objax.zoo.resnet_v2.ResNet34 method) (objax.zoo.resnet_v2.ResNet50 method) (objax.zoo.resnet_v2.ResNetV2 method) (objax.zoo.rnn.RNN method) (objax.zoo.vgg.VGG19 method) (objax.zoo.wide_resnet.WideResNet method) (objax.zoo.wide_resnet.WideResNetGeneral method) (objax.zoo.wide_resnet.WRNBlock method)

A

Adam (class in objax.optimizer) analyze_dp() (in module objax.privacy.dpsgd) analyze_dp_sample_without_replacement() (in module objax.privacy.dpsgd) analyze_renyi() (in module objax.privacy.dpsgd) analyze_renyi_sample_without_replacement() (in module objax.privacy.dpsgd) append() (objax.nn.Sequential method) args_indexes() (in module objax.util)

assign() (objax.RandomState method) (objax.StateVar method) (objax.TrainRef method) (objax.TrainVar method) (objax.VarCollection method) auto_vars() (objax.Function static method) average_pool_2d() (in module objax.functional)

B

BaseState (class in objax) BaseVar (class in objax) batch_to_space2d() (in module objax.functional) BatchNorm (class in objax.nn)

BatchNorm0D (class in objax.nn) BatchNorm1D (class in objax.nn) BatchNorm2D (class in objax.nn) build() (objax.zoo.vgg.VGG19 method)

C

celu() (in module objax.functional) channel_to_space2d() (in module objax.functional) Checkpoint (class in objax.io) checkpoint_idx() (objax.io.Checkpoint static method) clear() (objax.nn.Sequential method) (objax.util.EasyDict method) close() (objax.jaxboard.SummaryWriter method) Conv2D (class in objax.nn)

convert_renyidp_to_dp() (in module objax.privacy.dpsgd) ConvNet (class in objax.zoo.convnet) ConvPadding (class in objax) ConvTranspose2D (class in objax.nn) copy() (objax.nn.Sequential method) (objax.util.EasyDict method) count() (objax.nn.Sequential method) cross_entropy_logits() (in module objax.functional.loss) cross_entropy_logits_sparse() (in module objax.functional.loss)

D

DIR_NAME (objax.io.Checkpoint attribute) DNNet (class in objax.zoo.dnnet) Dropout (class in objax.nn) dtype (objax.RandomState property) (objax.StateVar property) (objax.TrainRef property) (objax.TrainVar property)

dummy_context_mgr() (in module objax.util) dynamic_slice() (in module objax.functional)

E

EasyDict (class in objax.util) elu() (in module objax.functional) ExponentialMovingAverage (class in objax.nn) (class in objax.optimizer)

ExponentialMovingAverageModule (class in objax.optimizer) extend() (objax.nn.Sequential method)

F

FILE_FORMAT (objax.io.Checkpoint attribute) FILE_MATCH (objax.io.Checkpoint attribute) find_used_variables() (in module objax.util) flatten() (in module objax.functional)

ForceArgs (class in objax) ForceArgs.ANY (class in objax) from_file() (in module objax.util.image) fromkeys() (objax.util.EasyDict method) Function (class in objax)

G

gain_leaky_relu (class in objax.nn.init) Generator (class in objax.random)

get() (objax.util.EasyDict method) Grad (class in objax) GradValues (class in objax)

H

Hessian (class in objax)

I

identity (class in objax.nn.init) ilog2() (in module objax.util) image() (objax.jaxboard.Summary method) image_grid() (in module objax.util.image)

index() (objax.nn.Sequential method) init_state() (objax.zoo.rnn.RNN method) insert() (objax.nn.Sequential method) interpolate() (in module objax.functional) items() (objax.util.EasyDict method)

J

Jacobian (class in objax)

Jit (class in objax)

K

kaiming_normal (class in objax.nn.init) kaiming_normal_gain (class in objax.nn.init) kaiming_truncated_normal (class in objax.nn.init)

key (objax.random.Generator property) keys() (objax.util.EasyDict method) kl() (in module objax.functional.divergence)

L

l2() (in module objax.functional.loss) LARS (class in objax.optimizer) leaky_relu() (in module objax.functional) Linear (class in objax.nn)

LOAD_FN() (objax.io.Checkpoint static method) load_var_collection() (in module objax.io) log_sigmoid() (in module objax.functional) log_softmax() (in module objax.functional) logsumexp() (in module objax.functional)

M

max_pool_2d() (in module objax.functional) mean_absolute_error() (in module objax.functional.loss) mean_reduce() (objax.zoo.wide_resnet.WideResNetGeneral static method) mean_squared_error() (in module objax.functional.loss) module objax.privacy.dpsgd objax.util.image

Module (class in objax) ModuleList (class in objax) Momentum (class in objax.optimizer) MovingAverage (class in objax.nn)

N

nchw() (in module objax.util.image) ndim (objax.RandomState property) (objax.StateVar property) (objax.TrainRef property) (objax.TrainVar property)

nhwc() (in module objax.util.image) normal() (in module objax.random) normalize_to_uint8() (in module objax.util.image) normalize_to_unit_float() (in module objax.util.image)

O

objax.privacy.dpsgd module objax.util.image module

Objax2Tf (class in objax.util) one_hot() (in module objax.functional) orthogonal (class in objax.nn.init)

P

pad() (in module objax.functional) Parallel (class in objax) pmax() (in module objax.functional.parallel) pmean() (in module objax.functional.parallel) pmin() (in module objax.functional.parallel)

pop() (objax.nn.Sequential method) (objax.util.EasyDict method) popitem() (objax.util.EasyDict method) positional_args_names() (in module objax.util) PrivateGradValues (class in objax.privacy.dpsgd) psum() (in module objax.functional.parallel)

R

randint() (in module objax.random) RandomState (class in objax) reduce() (objax.RandomState method) (objax.StateVar method) (objax.TrainRef method) (objax.TrainVar method) refs_and_values() (objax.optimizer.ExponentialMovingAverage method) relu() (in module objax.functional) remove() (objax.nn.Sequential method) rename() (objax.VarCollection method) Renamer (class in objax.util) replace_vars() (objax.optimizer.ExponentialMovingAverage method)

replicate() (objax.VarCollection method) reshape_microbatch() (objax.privacy.dpsgd.PrivateGradValues method) ResNet101 (class in objax.zoo.resnet_v2) ResNet152 (class in objax.zoo.resnet_v2) ResNet18 (class in objax.zoo.resnet_v2) ResNet200 (class in objax.zoo.resnet_v2) ResNet34 (class in objax.zoo.resnet_v2) ResNet50 (class in objax.zoo.resnet_v2) ResNetV2 (class in objax.zoo.resnet_v2) restore() (objax.io.Checkpoint method) reverse() (objax.nn.Sequential method) RNN (class in objax.zoo.rnn) rsqrt() (in module objax.functional)

S

SAME (objax.ConvPadding attribute) save() (objax.io.Checkpoint method) SAVE_FN() (objax.io.Checkpoint static method) save_var_collection() (in module objax.io) scalar() (objax.jaxboard.Summary method) scan() (in module objax.functional) seed() (objax.random.Generator method) (objax.RandomState method) selu() (in module objax.functional) Sequential (class in objax.nn) setdefault() (objax.util.EasyDict method) SGD (class in objax.optimizer) shape (objax.RandomState property) (objax.StateVar property) (objax.TrainRef property) (objax.TrainVar property)

sigmoid() (in module objax.functional) sigmoid_cross_entropy_logits() (in module objax.functional.loss) softmax() (in module objax.functional) softplus() (in module objax.functional) space_to_batch2d() (in module objax.functional) space_to_channel2d() (in module objax.functional) split() (objax.RandomState method) StateVar (class in objax) stop_gradient() (in module objax.functional) subset() (objax.VarCollection method) Summary (class in objax.jaxboard) SummaryWriter (class in objax.jaxboard) SyncedBatchNorm (class in objax.nn) SyncedBatchNorm0D (class in objax.nn) SyncedBatchNorm1D (class in objax.nn) SyncedBatchNorm2D (class in objax.nn)

T

tanh() (in module objax.functional) tensors() (objax.VarCollection method) text() (objax.jaxboard.Summary method) to_png() (in module objax.util.image) to_tuple() (in module objax.util)

top_k() (in module objax.functional) TrainRef (class in objax) TrainVar (class in objax) truncated_normal (class in objax.nn.init) truncated_normal() (in module objax.random)

U

undo() (objax.ForceArgs static method) uniform() (in module objax.random) update() (objax.util.EasyDict method) (objax.VarCollection method)

update_ema() (objax.optimizer.ExponentialMovingAverageModule method) upsample_2d() (in module objax.functional) upscale_nn() (in module objax.functional)

V

VALID (objax.ConvPadding attribute) value (objax.RandomState property) (objax.StateVar property) (objax.TrainRef property) (objax.TrainVar property) values() (objax.util.EasyDict method) VarCollection (class in objax) vars() (objax.ForceArgs method) (objax.Function method) (objax.Jit method) (objax.Module method) (objax.ModuleList method) (objax.nn.Sequential method) (objax.Parallel method) (objax.Vectorize method)

Vectorize (class in objax) VGG19 (class in objax.zoo.vgg)

W

WideResNet (class in objax.zoo.wide_resnet) WideResNetGeneral (class in objax.zoo.wide_resnet)

with_vars() (objax.Function static method) write() (objax.jaxboard.SummaryWriter method) WRNBlock (class in objax.zoo.wide_resnet)

X

xavier_normal (class in objax.nn.init)

xavier_normal_gain (class in objax.nn.init) xavier_truncated_normal (class in objax.nn.init)