.gitignore
LICENSE
README.md
pyproject.toml
aftab/Aftab.py
aftab/__init__.py
aftab/__main__.py
aftab/common.py
aftab/constants.py
aftab.egg-info/PKG-INFO
aftab.egg-info/SOURCES.txt
aftab.egg-info/dependency_links.txt
aftab.egg-info/requires.txt
aftab.egg-info/top_level.txt
aftab/encoders/AlphaEncoder.py
aftab/encoders/BetaEncoder.py
aftab/encoders/DeltaEncoder.py
aftab/encoders/EpsilonEncoder.py
aftab/encoders/EtaEncoder.py
aftab/encoders/GammaEncoder.py
aftab/encoders/HadamaxDeltaEncoder.py
aftab/encoders/HadamaxEpsilonEncoder.py
aftab/encoders/HadamaxGammaEncoderV1.py
aftab/encoders/HadamaxGammaEncoderV2.py
aftab/encoders/HadamaxNatureDQNEncoder.py
aftab/encoders/HadamaxZetaEncoder.py
aftab/encoders/NatureDQNEncoder.py
aftab/encoders/ThetaEncoder.py
aftab/encoders/ZetaEncoder.py
aftab/encoders/__init__.py
aftab/exceptions/__init__.py
aftab/functions/__init__.py
aftab/functions/epsilon_greedy_vectorized.py
aftab/functions/flush.py
aftab/functions/lambda_returns.py
aftab/functions/mse_loss.py
aftab/jax/AftabJax.py
aftab/jax/__init__.py
aftab/jax/archive.py
aftab/jax/encoders/AlphaEncoder.py
aftab/jax/encoders/BaseEncoder.py
aftab/jax/encoders/BetaEncoder.py
aftab/jax/encoders/NatureDQNEncoder.py
aftab/jax/encoders/__init__.py
aftab/jax/modules/Conv2d.py
aftab/jax/modules/EncoderBlock.py
aftab/jax/modules/LayerNorm2d.py
aftab/jax/modules/Linear.py
aftab/jax/modules/Stream.py
aftab/jax/modules/__init__.py
aftab/jax/netwoks/QNetwork.py
aftab/maps/__init__.py
aftab/maps/acceptable_frames_map.py
aftab/maps/encoders_map.py
aftab/maps/networks_map.py
aftab/maps/optimizers_maps.py
aftab/mixins/AftabActionsMixin.py
aftab/mixins/AftabBaseMixin.py
aftab/mixins/AftabEnvironmentMixin.py
aftab/mixins/AftabLossMixin.py
aftab/mixins/AftabNetworkMixin.py
aftab/mixins/AftabOptimizerMixin.py
aftab/mixins/AftabQValueMixin.py
aftab/mixins/AftabReturnsMixin.py
aftab/mixins/AftabStoreModelMixin.py
aftab/mixins/AftabTrainMixin.py
aftab/mixins/AftabTrainingResultsMixin.py
aftab/mixins/__init__.py
aftab/modules/EncoderBlock.py
aftab/modules/HadamaxBlock.py
aftab/modules/HadamaxLayerNorm2d.py
aftab/modules/LayerNorm2d.py
aftab/modules/Stream.py
aftab/modules/__init__.py
aftab/networks/BaseNetwork.py
aftab/networks/BootstrappedDuellingNetwork.py
aftab/networks/BootstrappedNetwork.py
aftab/networks/DistributionalBootstrappedDuellingNetwork.py
aftab/networks/DistributionalDuellingNetwork.py
aftab/networks/DistributionalNetwork.py
aftab/networks/DuellingNetwork.py
aftab/networks/PQNNetwork.py
aftab/networks/__init__.py
examples/run_experiments.py
figures/header.svg
figures/encoder_experiments/global.png
figures/encoder_experiments/global_zoomed.png
figures/encoder_experiments/statistical_significance.png
figures/final_experiments/global.png
figures/final_experiments/global_zoomed.png
figures/hadamax_experiments/global.png
figures/hadamax_experiments/global_zoomed.png
figures/hadamax_experiments/statistical_significance.png
results/encoder_experiments/human_normalized_scores.md
results/encoder_experiments/scores.md
results/final_experiments/human_normalized_scores.md
results/final_experiments/scores.md
results/hadamax_experiments/human_normalized_scores.md
results/hadamax_experiments/scores.md
scripts/README.md
scripts/install_micromamba.sh
scripts/setup_environment.sh
scripts/submit_slurm.sh