memorax/__init__.py,sha256=kQ-fDiVlseJ6FGjAw1YcHvTVTJFjebE_5SkCM9dC3mU,1289
memorax/algorithms/__init__.py,sha256=BES1AzZu1uj-iAaj3Cmb0fpiVOkwF66TpstOye_FdVw,168
memorax/algorithms/dqn.py,sha256=S_PzrX1vV4R95OuGbF01j0qm0EWQavOEOtkxRs1EvNs,12525
memorax/algorithms/ppo.py,sha256=Bn0YMG2Vb5HLUjmXZufKSOCFs9wtfrn24G8wYA_kLKc,21437
memorax/algorithms/pqn.py,sha256=pj6erxmmaV_vJfZ5kSiB2irruLiKx7Ok5uFtH_l2t4k,14078
memorax/algorithms/sac.py,sha256=vVrMs0n6EdkjlNrCnxbBUdJi2R1C1qNeOq9rr3AvBBg,15401
memorax/buffers/__init__.py,sha256=JI2vIuBz5TABqA-O6107RXqlJZXG8xpssbPyzVeihBI,128
memorax/buffers/episode_buffer.py,sha256=BnFzawKbhBQlFaSNV5acepnw-CYRB_RE5VLoKFQ4VtU,7338
memorax/environments/__init__.py,sha256=QpfekPJqKHZJZpAo7si3oCEkf9-du0t3o25RRJ7K96Q,50
memorax/environments/brax.py,sha256=B4RxuC_yIZCl_OZ7nzUUUcZEi8-ajpcjWLiPDevFyTQ,2292
memorax/environments/craftax.py,sha256=njAGeZf4nDNmFogLAs9mHJ28nmnc1KCNCBRTkdl529M,2176
memorax/environments/environment.py,sha256=rnphGukhAl7yFz5WJ2v1jl3vti2u8Pqun962g1e8vHw,684
memorax/environments/gxm.py,sha256=UICOOVVWPD3DvOa3HC_nYrSasZ9LrGS2Mk6rsy2HU6M,2442
memorax/environments/gymnax.py,sha256=LGpErfj9lat5UfbGueMNSEz_qI5gPtdr34BBGSa2udQ,1805
memorax/environments/memory_rl.py,sha256=INb3hZPuZehvUra2SXyGBfs6qhNThep337Ywu7vj1o8,348
memorax/environments/mujoco.py,sha256=bcCx6RA0rfbxKWVIGheAOoQtFkSi-_ldpiUJDOt5Twc,2969
memorax/environments/navix.py,sha256=GgRrYsA03-rQ1jL-sWoZ1mJTGz4c8VxA_yi0qktNzPE,1064
memorax/environments/popgym_arcade.py,sha256=rd0hTIxTjDHcfXHbG_tbMDBLzFYk5GlYZw2dGA94xOU,743
memorax/environments/popjym.py,sha256=33yuIfexSZWbqSzCNODGun8C5keIfFzskpEisfCz6VI,1839
memorax/environments/xminigrid.py,sha256=O58rDlWWJCrQNQD5OOiDSJvClNBOPJqvaxrEyfQbcX4,1841
memorax/kernels/__init__.py,sha256=GXXuXtkBWUU-f9ur2-OQeNt3ZD0U9-nc-8GvdcsadxM,34
memorax/kernels/linear_attention.py,sha256=WeVXUhu8ojXm_CGPImJnbo8xqdq9b2_nCynsn_FQUss,19627
memorax/loggers/__init__.py,sha256=VE_WAhkTcOTUzF6WxtAl2VyN-Y46IV7BRVw6fcKA21w,357
memorax/loggers/console.py,sha256=kQVt0eYArbwS6Z5QN02fz8Pf2xmkeuzjD6f1TICuo2E,2718
memorax/loggers/dashboard.py,sha256=06YW28C8CXFVC7-4uMp2Wkq7EGX48VxSvBrPzjXqThI,6894
memorax/loggers/file.py,sha256=2ejZuZqAOUjC76a9dP60-fprQbBRxODHXRqQ2WANlD0,3290
memorax/loggers/logger.py,sha256=5ifbKX8AKJCr93CcbrldQ_V3JP-FDVR6CS6QNrD6CUo,2852
memorax/loggers/neptune.py,sha256=GXEflI7u_KB6CYc7vlV56zwW71UsZQSbzDX4A83xbjE,1605
memorax/loggers/tensorboard.py,sha256=cn6_-ZN7n_RLJM2aGMeCH3jVtH53WfzS4xnXyBk5x2M,1582
memorax/loggers/wandb.py,sha256=28ODdO1f-KS7Eiam2aSjLH6KciFA-g-IREmv2NCUgHw,2320
memorax/networks/__init__.py,sha256=CEnn-BatmPZJkY4olZqAjKFByQfGBtvvBPhW1HFwqPA,979
memorax/networks/cnn.py,sha256=3U0TdBwWyfm91kcm9Oe0MwJqmEBtcwxmseLda6CZBOk,1445
memorax/networks/embedding.py,sha256=cfwhGVtl0eGjR3ZWfZ1cm_UYHecphfJwxVUlNdAQ3XE,560
memorax/networks/feature_extractor.py,sha256=Yk2WNS9aL54x78e5cRi8eWXRDFIX-_ip-apjl3loDSI,1068
memorax/networks/heads.py,sha256=wHEfWLsCr5L8_zwH_zzUYs3sPcm2lM4sIiXYVmaw98k,4023
memorax/networks/identity.py,sha256=EV-gNdOU4cUsCMzeyLGHwisSyO-wlFUub-GVfHNwZcA,130
memorax/networks/mlp.py,sha256=NhGJJYIzqyKk3w0G5IYbSInCnnVrwTK2xk8TQfcT2f4,921
memorax/networks/network.py,sha256=e57i4zuqUDi1RGi3U0tDXeipEMXU0HXEREHCzwdS4MY,937
memorax/networks/vit.py,sha256=WKYgawuToHmTte10A7DADXG6eK9nS1pUjlN5yO1IuO0,1537
memorax/networks/blocks/__init__.py,sha256=OqPQNfg6pffnHFM8o_o_k6ZwwuZWipwDKhIO3AGPYSI,273
memorax/networks/blocks/base.py,sha256=Oftk6l9-QggXCYGzIt-SSE1ssgigz_kbFmUPOWdcLWo,586
memorax/networks/blocks/ffn.py,sha256=glf6iKsRjVrGxI9rO60z2HXtzkhp5xQAE1iRTTpeFBk,2611
memorax/networks/blocks/moe.py,sha256=qal_igilqLjRWHs_hTJTYCrMkQe4x18X4zlG095bH1Q,1756
memorax/networks/blocks/normalization.py,sha256=KE5QeVc2lIHWd5hiaMTyHN5WlOvvviHIvHAAz77GH-I,1773
memorax/networks/blocks/residual.py,sha256=q-rcsh1O6oS8fRc-Q-4wG-A8pBa1rGsfDQhcI6WyOfE,1590
memorax/networks/blocks/router.py,sha256=aLVGOGAqczdg58FVXjRFFj0V02ik4yWbpKKwGSSLBjM,1136
memorax/networks/blocks/segment_recurrence.py,sha256=F-B7Ux_e8LYfuO-rS17xZzOc0tkgD2XC-ojPj66mbno,2523
memorax/networks/blocks/stack.py,sha256=0rEbdEjce3_NIuLt9RAHpHzFqG36cgc4mvjewNOjDoc,1807
memorax/networks/positional_embeddings/__init__.py,sha256=KY9QiVclrRWONQ4tn2Thzs_ciISppTxHAuotO-rvYKU,169
memorax/networks/positional_embeddings/alibi.py,sha256=xzhqv3yVu6cez4KsRjBoxAP9HGTRkXAK5BBBQHrVc8g,958
memorax/networks/positional_embeddings/base.py,sha256=yARkKMBmnoqYpOvKv0JT6K2M4WrfltX614jp5FG-vOU,466
memorax/networks/positional_embeddings/rope.py,sha256=LZ3psgWhSwz74Jy4EJRKCiSUJkOOhVJBJGJU2WTnV0s,1462
memorax/networks/positional_embeddings/wpe.py,sha256=dzdmcOOj8YFzFbqDzxHS2PsIOuo5CvW-oYkfZp0IC9U,1870
memorax/networks/sequence_models/__init__.py,sha256=o3VjLQldb7PYHlHqh0Ph3iS9_HynsmUtMbaH2pEjJVI,481
memorax/networks/sequence_models/ffm.py,sha256=IJMWlGQ1AIBa34lUQgsiAqQLW6-bZ4Scg8vQFTgSEDo,4950
memorax/networks/sequence_models/linear_attention.py,sha256=oVlnxXEgpOkeWYoFyTqXAbC6ojvWv0-B0geKZjxU1c0,4379
memorax/networks/sequence_models/lru.py,sha256=0RLiA0ey9qDtDjvBytNbm9RAgy0G02TzVnuUFb3kuBs,3942
memorax/networks/sequence_models/mamba.py,sha256=w6_rI_5MLE3mGmfTrDfmNvbFAC_cWI-fa8oXCuQ4w6w,6068
memorax/networks/sequence_models/memoroid.py,sha256=VNrMYkVjJfwK9El_e0xPygoDx3IWzsWAUHE1jcnXuKk,2238
memorax/networks/sequence_models/min_gru.py,sha256=F748jqxC6JgDjqioC0nXNiZOTskI_vbHtR8UfgDJrNo,2573
memorax/networks/sequence_models/mlstm.py,sha256=kG4RiY-94P6e4fTYMI7TLVy4WHW8m-vycntad9JbjBo,9548
memorax/networks/sequence_models/rnn.py,sha256=hbkaCoFdhCHrirw3_s0PVg0bgAnAnVK6Kb-SOiYGKdI,2161
memorax/networks/sequence_models/s5.py,sha256=sS1ARQSvcFXw-4IwFWS9o90BozGcf8GGv4slmkEdubk,5639
memorax/networks/sequence_models/self_attention.py,sha256=NfalgYI9mleXqKYhZU63kAJCcEpT6_-KuWicOij7ppI,4217
memorax/networks/sequence_models/sequence_model.py,sha256=_2lgZBQN0O01Cg5SXU2Lq93_edpWXYTRGk87BRWIt0U,439
memorax/networks/sequence_models/shm.py,sha256=48HsYmSmLWS67OIhKKsTpmOFSXld7yWqK_RNZeIdF34,3164
memorax/networks/sequence_models/slstm.py,sha256=t5eN_mpolrD0CpPAc4NchKJbnyd1NNomYuif_hbmC3U,6987
memorax/networks/sequence_models/utils.py,sha256=N3yVHpqJz7vRMiDVZYQ5y-OKcpRr530tU2_9Ol-Bc08,11970
memorax/networks/sequence_models/wrappers.py,sha256=34x-zu3iwLF4LuxHVRAs7MvelM1yMO9P0sd_0_wsiDc,1591
memorax/optimizers/transforms/apply_if.py,sha256=I1wLVcMi4bCyDAN7FikbpRBcfo3d5DfZ828beSkTtaI,1342
memorax/utils/__init__.py,sha256=tRlVSY4rWfr0UHiOGXsKGMIaEI6s3t3HZi3_t-LWbws,215
memorax/utils/decorators.py,sha256=zh5sgBUPgwzfHBCufOmQaMV8mKLxBUMHGza0spfwkGI,180
memorax/utils/gae.py,sha256=e4f__0228nvqgbHebWaQt7UYoly1l141s9Icdrxfkcs,798
memorax/utils/stats.py,sha256=tQdU-kvZQUx_2ju2BfwUt6R6lSL-ejOXZIBymHXl2xI,542
memorax/utils/timestep.py,sha256=Ws3-9zyNdGo9fYYPdTtNRWlSfTz-g3UDxzt6O18JENM,511
memorax/utils/transition.py,sha256=8QWyByTQp8baZiC1SM9jRBjkSxu8HsF_YxgikEvK5mo,2145
memorax/utils/typing.py,sha256=bap9Uw8egfx7KLgpe8zXpiv2oc-lHwchTvDMb9T1MQg,1177
memorax/utils/update.py,sha256=IXkh721ZVs4m0soz2UXnuHbHUc9haLJVYHY9D8OEca0,2211
memorax/utils/utils.py,sha256=k_M0hFbgTiEO6SHVQeMlL-MQkC4XlXWJ7o7mZ_2LhVs,250
memorax/utils/wrappers.py,sha256=x8iAyNRe9slhvUlla37XOjCN04ykd3VER332ySdOnrI,5086
memorax-1.0.0.dist-info/licenses/LICENSE,sha256=gEG9oKxMEF2pya5FsRQDim8yyoincn9K0LDb8D74e0Q,10758
memorax-1.0.0.dist-info/METADATA,sha256=kKF87MapmOcLL_m_qHwP4XQk1O7tbOSjpFrxsjiQ1Uc,7179
memorax-1.0.0.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
memorax-1.0.0.dist-info/top_level.txt,sha256=QGu5078aYSEU4uZno8n5P7TipLLtXqx2FQ6bWWetNjc,8
memorax-1.0.0.dist-info/RECORD,,
