Objax
v1.0.2
Getting Started
Installation and Setup
Objax Basics Tutorial
Logistic Regression
Creating Custom Networks for Multi-Class Classification
Code Examples
API documentation
Objax API
In-depth topics
Variables and Modules
Understanding Gradients
Compilation and Parallelism
Loading and Saving
Objax
Docs
»
Index
Edit on GitHub
Index
_
|
A
|
B
|
C
|
D
|
E
|
F
|
G
|
I
|
J
|
K
|
L
|
M
|
N
|
O
|
P
|
R
|
S
|
T
|
U
|
V
|
W
|
X
_
__call__() (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.Momentum method)
(objax.optimizer.SGD method)
(objax.Parallel method)
(objax.privacy.PrivateGradValues method)
(objax.random.Generator method)
(objax.Vectorize method)
(objax.zoo.rnn.RNN method)
(objax.zoo.wide_resnet.WRNBlock method)
__init__() (objax.BaseState method)
(objax.BaseVar method)
(objax.GradValues method)
(objax.io.Checkpoint 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.Momentum method)
(objax.optimizer.SGD method)
(objax.Parallel method)
(objax.privacy.PrivateGradValues method)
(objax.random.Generator method)
(objax.RandomState method)
(objax.StateVar method)
(objax.TrainRef method)
(objax.TrainVar method)
(objax.util.EasyDict 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)
append() (objax.nn.Sequential method)
apply_dp_sgd_analysis() (in module objax.privacy)
args_indexes() (in module objax.util)
assign() (objax.RandomState method)
(objax.StateVar method)
(objax.TrainRef method)
(objax.TrainVar method)
(objax.VarCollection 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)
compute_rdp() (in module objax.privacy)
Conv2D (class in objax.nn)
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)
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)
extend() (objax.nn.Sequential method)
F
FILE_FORMAT (objax.io.Checkpoint attribute)
FILE_MATCH (objax.io.Checkpoint attribute)
flatten() (in module objax.functional)
fromkeys() (objax.util.EasyDict method)
G
gain_leaky_relu (class in objax.nn.init)
Generator (class in objax.random)
get() (objax.util.EasyDict method)
get_privacy_spent() (in module objax.privacy)
GradValues (class in objax)
I
ilog2() (in module objax.util)
image() (objax.jaxboard.Summary method)
index() (objax.nn.Sequential method)
init_state() (objax.zoo.rnn.RNN method)
insert() (objax.nn.Sequential method)
items() (objax.util.EasyDict method)
J
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)
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_reduce() (objax.zoo.wide_resnet.WideResNetGeneral static method)
module
objax.privacy
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)
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
module
objax.util.image
module
one_hot() (in module objax.functional)
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)
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)
replace_vars() (objax.optimizer.ExponentialMovingAverage method)
replicate() (objax.VarCollection method)
reshape_microbatch() (objax.privacy.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)
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)
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
uniform() (in module objax.random)
update() (objax.util.EasyDict method)
(objax.VarCollection method)
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.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)
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)
Read the Docs
v: v1.0.2
Versions
latest
stable
v1.0.2
Downloads
On Read the Docs
Project Home
Builds
Free document hosting provided by
Read the Docs
.