LICENSE
MANIFEST.in
README.md
requirements.txt
setup.py
objax/__init__.py
objax/_patch_jax.py
objax/_version.py
objax/constants.py
objax/gradient.py
objax/jaxboard.py
objax/module.py
objax/typing.py
objax/variable.py
objax.egg-info/PKG-INFO
objax.egg-info/SOURCES.txt
objax.egg-info/dependency_links.txt
objax.egg-info/requires.txt
objax.egg-info/top_level.txt
objax/functional/__init__.py
objax/functional/divergence.py
objax/functional/loss.py
objax/functional/parallel.py
objax/functional/core/__init__.py
objax/functional/core/activation.py
objax/functional/core/ops.py
objax/functional/core/pooling.py
objax/io/__init__.py
objax/io/checkpoint.py
objax/io/ops.py
objax/nn/__init__.py
objax/nn/init.py
objax/nn/layers.py
objax/optimizer/__init__.py
objax/optimizer/adam.py
objax/optimizer/ema.py
objax/optimizer/lars.py
objax/optimizer/momentum.py
objax/optimizer/scheduler.py
objax/optimizer/sgd.py
objax/privacy/__init__.py
objax/privacy/dpsgd/__init__.py
objax/privacy/dpsgd/gradient.py
objax/privacy/dpsgd/privacyaccountant.py
objax/random/__init__.py
objax/random/random.py
objax/util/__init__.py
objax/util/check.py
objax/util/image.py
objax/util/objax2tf.py
objax/util/tracing.py
objax/util/util.py
objax/zoo/__init__.py
objax/zoo/convnet.py
objax/zoo/dnnet.py
objax/zoo/resnet_v2.py
objax/zoo/rnn.py
objax/zoo/vgg.py
objax/zoo/wide_resnet.py