# Autogenerated by nbdev

d = { 'settings': { 'branch': 'main',
                'doc_baseurl': '/hamux',
                'doc_host': 'https://bhoov.github.io',
                'git_url': 'https://github.com/bhoov/hamux',
                'lib_path': 'hamux'},
  'syms': { 'hamux.core': {'hamux.core.foo': ('core.html#foo', 'hamux/core.py')},
            'hamux.datasets': {},
            'hamux.ham': { 'hamux.ham.HAM': ('ham.html#ham', 'hamux/ham.py'),
                           'hamux.ham.HAM.__init__': ('ham.html#ham.__init__', 'hamux/ham.py'),
                           'hamux.ham.HAM._statelist_batch_axes': ('ham.html#ham._statelist_batch_axes', 'hamux/ham.py'),
                           'hamux.ham.HAM.activations': ('ham.html#ham.activations', 'hamux/ham.py'),
                           'hamux.ham.HAM.alphas': ('ham.html#ham.alphas', 'hamux/ham.py'),
                           'hamux.ham.HAM.dEdg': ('ham.html#ham.dedg', 'hamux/ham.py'),
                           'hamux.ham.HAM.energy': ('ham.html#ham.energy', 'hamux/ham.py'),
                           'hamux.ham.HAM.init_states': ('ham.html#ham.init_states', 'hamux/ham.py'),
                           'hamux.ham.HAM.init_states_and_params': ('ham.html#ham.init_states_and_params', 'hamux/ham.py'),
                           'hamux.ham.HAM.layer_energy': ('ham.html#ham.layer_energy', 'hamux/ham.py'),
                           'hamux.ham.HAM.layer_taus': ('ham.html#ham.layer_taus', 'hamux/ham.py'),
                           'hamux.ham.HAM.load_ckpt': ('ham.html#ham.load_ckpt', 'hamux/ham.py'),
                           'hamux.ham.HAM.load_state_dict': ('ham.html#ham.load_state_dict', 'hamux/ham.py'),
                           'hamux.ham.HAM.n_connections': ('ham.html#ham.n_connections', 'hamux/ham.py'),
                           'hamux.ham.HAM.n_layers': ('ham.html#ham.n_layers', 'hamux/ham.py'),
                           'hamux.ham.HAM.n_synapses': ('ham.html#ham.n_synapses', 'hamux/ham.py'),
                           'hamux.ham.HAM.save_state_dict': ('ham.html#ham.save_state_dict', 'hamux/ham.py'),
                           'hamux.ham.HAM.step': ('ham.html#ham.step', 'hamux/ham.py'),
                           'hamux.ham.HAM.synapse_energy': ('ham.html#ham.synapse_energy', 'hamux/ham.py'),
                           'hamux.ham.HAM.updates': ('ham.html#ham.updates', 'hamux/ham.py'),
                           'hamux.ham.HAM.vactivations': ('ham.html#ham.vactivations', 'hamux/ham.py'),
                           'hamux.ham.HAM.vdEdg': ('ham.html#ham.vdedg', 'hamux/ham.py'),
                           'hamux.ham.HAM.venergy': ('ham.html#ham.venergy', 'hamux/ham.py'),
                           'hamux.ham.HAM.vupdates': ('ham.html#ham.vupdates', 'hamux/ham.py')},
            'hamux.lagrangians': { 'hamux.lagrangians.LExp': ('lagrangians.html#lexp', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians.LExp.__call__': ('lagrangians.html#lexp.__call__', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians.LExp.__init__': ('lagrangians.html#lexp.__init__', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians.LIdentity': ('lagrangians.html#lidentity', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians.LIdentity.__call__': ('lagrangians.html#lidentity.__call__', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians.LIdentity.__init__': ('lagrangians.html#lidentity.__init__', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians.LLayerNorm': ('lagrangians.html#llayernorm', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians.LLayerNorm.__call__': ( 'lagrangians.html#llayernorm.__call__',
                                                                              'hamux/lagrangians.py'),
                                   'hamux.lagrangians.LLayerNorm.__init__': ( 'lagrangians.html#llayernorm.__init__',
                                                                              'hamux/lagrangians.py'),
                                   'hamux.lagrangians.LRelu': ('lagrangians.html#lrelu', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians.LRelu.__call__': ('lagrangians.html#lrelu.__call__', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians.LRelu.__init__': ('lagrangians.html#lrelu.__init__', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians.LRepu': ('lagrangians.html#lrepu', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians.LRepu.__call__': ('lagrangians.html#lrepu.__call__', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians.LRepu.__init__': ('lagrangians.html#lrepu.__init__', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians.LRexp': ('lagrangians.html#lrexp', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians.LRexp.__call__': ('lagrangians.html#lrexp.__call__', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians.LRexp.__init__': ('lagrangians.html#lrexp.__init__', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians.LSigmoid': ('lagrangians.html#lsigmoid', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians.LSigmoid.__call__': ('lagrangians.html#lsigmoid.__call__', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians.LSigmoid.__init__': ('lagrangians.html#lsigmoid.__init__', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians.LSoftmax': ('lagrangians.html#lsoftmax', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians.LSoftmax.__call__': ('lagrangians.html#lsoftmax.__call__', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians.LSoftmax.__init__': ('lagrangians.html#lsoftmax.__init__', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians.LSphericalNorm': ('lagrangians.html#lsphericalnorm', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians.LSphericalNorm.__call__': ( 'lagrangians.html#lsphericalnorm.__call__',
                                                                                  'hamux/lagrangians.py'),
                                   'hamux.lagrangians.LSphericalNorm.__init__': ( 'lagrangians.html#lsphericalnorm.__init__',
                                                                                  'hamux/lagrangians.py'),
                                   'hamux.lagrangians.LTanh': ('lagrangians.html#ltanh', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians.LTanh.__call__': ('lagrangians.html#ltanh.__call__', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians.LTanh.__init__': ('lagrangians.html#ltanh.__init__', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians._lagr_sigmoid': ('lagrangians.html#_lagr_sigmoid', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians._lagr_sigmoid_jvp': ('lagrangians.html#_lagr_sigmoid_jvp', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians._lagr_tanh': ('lagrangians.html#_lagr_tanh', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians._lagr_tanh_defjvp': ('lagrangians.html#_lagr_tanh_defjvp', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians._simple_layernorm': ('lagrangians.html#_simple_layernorm', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians._simple_spherical_norm': ( 'lagrangians.html#_simple_spherical_norm',
                                                                                 'hamux/lagrangians.py'),
                                   'hamux.lagrangians._tempered_sigmoid': ('lagrangians.html#_tempered_sigmoid', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians.lagr_exp': ('lagrangians.html#lagr_exp', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians.lagr_identity': ('lagrangians.html#lagr_identity', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians.lagr_layernorm': ('lagrangians.html#lagr_layernorm', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians.lagr_relu': ('lagrangians.html#lagr_relu', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians.lagr_repu': ('lagrangians.html#lagr_repu', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians.lagr_rexp': ('lagrangians.html#lagr_rexp', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians.lagr_sigmoid': ('lagrangians.html#lagr_sigmoid', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians.lagr_softmax': ('lagrangians.html#lagr_softmax', 'hamux/lagrangians.py'),
                                   'hamux.lagrangians.lagr_spherical_norm': ( 'lagrangians.html#lagr_spherical_norm',
                                                                              'hamux/lagrangians.py'),
                                   'hamux.lagrangians.lagr_tanh': ('lagrangians.html#lagr_tanh', 'hamux/lagrangians.py')},
            'hamux.layers': { 'hamux.layers.Layer': ('layers.html#layer', 'hamux/layers.py'),
                              'hamux.layers.Layer.__call__': ('layers.html#layer.__call__', 'hamux/layers.py'),
                              'hamux.layers.Layer.__init__': ('layers.html#layer.__init__', 'hamux/layers.py'),
                              'hamux.layers.Layer.activation': ('layers.html#layer.activation', 'hamux/layers.py'),
                              'hamux.layers.Layer.energy': ('layers.html#layer.energy', 'hamux/layers.py'),
                              'hamux.layers.Layer.g': ('layers.html#layer.g', 'hamux/layers.py'),
                              'hamux.layers.Layer.init_state': ('layers.html#layer.init_state', 'hamux/layers.py'),
                              'hamux.layers.MakeLayer': ('layers.html#makelayer', 'hamux/layers.py')},
            'hamux.registry': { 'hamux.registry.conv_ham': ('registry.html#conv_ham', 'hamux/registry.py'),
                                'hamux.registry.conv_ham_avgpool_cifar': ('registry.html#conv_ham_avgpool_cifar', 'hamux/registry.py'),
                                'hamux.registry.conv_ham_avgpool_mnist': ('registry.html#conv_ham_avgpool_mnist', 'hamux/registry.py'),
                                'hamux.registry.conv_ham_maxpool_cifar': ('registry.html#conv_ham_maxpool_cifar', 'hamux/registry.py'),
                                'hamux.registry.conv_ham_maxpool_mnist': ('registry.html#conv_ham_maxpool_mnist', 'hamux/registry.py'),
                                'hamux.registry.create_model': ('registry.html#create_model', 'hamux/registry.py'),
                                'hamux.registry.energy_attn': ('registry.html#energy_attn', 'hamux/registry.py'),
                                'hamux.registry.energy_attn_cifar': ('registry.html#energy_attn_cifar', 'hamux/registry.py'),
                                'hamux.registry.energy_attn_mnist': ('registry.html#energy_attn_mnist', 'hamux/registry.py'),
                                'hamux.registry.fwd_conv': ('registry.html#fwd_conv', 'hamux/registry.py'),
                                'hamux.registry.fwd_vec': ('registry.html#fwd_vec', 'hamux/registry.py'),
                                'hamux.registry.hn': ('registry.html#hn', 'hamux/registry.py'),
                                'hamux.registry.hn_relu_cifar': ('registry.html#hn_relu_cifar', 'hamux/registry.py'),
                                'hamux.registry.hn_relu_mnist': ('registry.html#hn_relu_mnist', 'hamux/registry.py'),
                                'hamux.registry.hn_repu5_cifar': ('registry.html#hn_repu5_cifar', 'hamux/registry.py'),
                                'hamux.registry.hn_repu5_mnist': ('registry.html#hn_repu5_mnist', 'hamux/registry.py'),
                                'hamux.registry.hn_softmax_cifar': ('registry.html#hn_softmax_cifar', 'hamux/registry.py'),
                                'hamux.registry.hn_softmax_mnist': ('registry.html#hn_softmax_mnist', 'hamux/registry.py'),
                                'hamux.registry.named_partial': ('registry.html#named_partial', 'hamux/registry.py'),
                                'hamux.registry.register_model': ('registry.html#register_model', 'hamux/registry.py'),
                                'hamux.registry.simple_fwd': ('registry.html#simple_fwd', 'hamux/registry.py')},
            'hamux.synapses': { 'hamux.synapses.AttentionSynapse': ('synapses.html#attentionsynapse', 'hamux/synapses.py'),
                                'hamux.synapses.AttentionSynapse.__call__': ( 'synapses.html#attentionsynapse.__call__',
                                                                              'hamux/synapses.py'),
                                'hamux.synapses.AttentionSynapse.__init__': ( 'synapses.html#attentionsynapse.__init__',
                                                                              'hamux/synapses.py'),
                                'hamux.synapses.BinaryMixerSynapse': ('synapses.html#binarymixersynapse', 'hamux/synapses.py'),
                                'hamux.synapses.BinaryMixerSynapse.__call__': ( 'synapses.html#binarymixersynapse.__call__',
                                                                                'hamux/synapses.py'),
                                'hamux.synapses.BinaryMixerSynapse.__init__': ( 'synapses.html#binarymixersynapse.__init__',
                                                                                'hamux/synapses.py'),
                                'hamux.synapses.ConvSynapse': ('synapses.html#convsynapse', 'hamux/synapses.py'),
                                'hamux.synapses.ConvSynapse.__call__': ('synapses.html#convsynapse.__call__', 'hamux/synapses.py'),
                                'hamux.synapses.ConvSynapse.__init__': ('synapses.html#convsynapse.__init__', 'hamux/synapses.py'),
                                'hamux.synapses.ConvSynapse.example_output': ( 'synapses.html#convsynapse.example_output',
                                                                               'hamux/synapses.py'),
                                'hamux.synapses.ConvSynapseWithPool': ('synapses.html#convsynapsewithpool', 'hamux/synapses.py'),
                                'hamux.synapses.ConvSynapseWithPool.__call__': ( 'synapses.html#convsynapsewithpool.__call__',
                                                                                 'hamux/synapses.py'),
                                'hamux.synapses.ConvSynapseWithPool.__init__': ( 'synapses.html#convsynapsewithpool.__init__',
                                                                                 'hamux/synapses.py'),
                                'hamux.synapses.ConvSynapseWithPool.example_output': ( 'synapses.html#convsynapsewithpool.example_output',
                                                                                       'hamux/synapses.py'),
                                'hamux.synapses.DenseMatrixSynapseWithHiddenLayer': ( 'synapses.html#densematrixsynapsewithhiddenlayer',
                                                                                      'hamux/synapses.py'),
                                'hamux.synapses.DenseMatrixSynapseWithHiddenLayer.__call__': ( 'synapses.html#densematrixsynapsewithhiddenlayer.__call__',
                                                                                               'hamux/synapses.py'),
                                'hamux.synapses.DenseMatrixSynapseWithHiddenLayer.__init__': ( 'synapses.html#densematrixsynapsewithhiddenlayer.__init__',
                                                                                               'hamux/synapses.py'),
                                'hamux.synapses.DenseSynapse': ('synapses.html#densesynapse', 'hamux/synapses.py'),
                                'hamux.synapses.DenseSynapse.__call__': ('synapses.html#densesynapse.__call__', 'hamux/synapses.py'),
                                'hamux.synapses.DenseSynapse.__init__': ('synapses.html#densesynapse.__init__', 'hamux/synapses.py'),
                                'hamux.synapses.DenseTensorSynapseWithHiddenLayer': ( 'synapses.html#densetensorsynapsewithhiddenlayer',
                                                                                      'hamux/synapses.py'),
                                'hamux.synapses.DenseTensorSynapseWithHiddenLayer.__call__': ( 'synapses.html#densetensorsynapsewithhiddenlayer.__call__',
                                                                                               'hamux/synapses.py'),
                                'hamux.synapses.DenseTensorSynapseWithHiddenLayer.__init__': ( 'synapses.html#densetensorsynapsewithhiddenlayer.__init__',
                                                                                               'hamux/synapses.py'),
                                'hamux.synapses.SelfAttentionSynapse': ('synapses.html#selfattentionsynapse', 'hamux/synapses.py'),
                                'hamux.synapses.SelfAttentionSynapse.__call__': ( 'synapses.html#selfattentionsynapse.__call__',
                                                                                  'hamux/synapses.py'),
                                'hamux.synapses.SimpleDenseSynapse': ('synapses.html#simpledensesynapse', 'hamux/synapses.py'),
                                'hamux.synapses.SimpleDenseSynapse.__call__': ( 'synapses.html#simpledensesynapse.__call__',
                                                                                'hamux/synapses.py'),
                                'hamux.synapses.Synapse': ('synapses.html#synapse', 'hamux/synapses.py'),
                                'hamux.synapses.Synapse.__call__': ('synapses.html#synapse.__call__', 'hamux/synapses.py'),
                                'hamux.synapses.Synapse.energy': ('synapses.html#synapse.energy', 'hamux/synapses.py')},
            'hamux.utils': {}}}