CMB_Topology_ML_JAX

Neural network layers and networks to study the topology of the universe. Based on graph convolutions and written in the jax environment

Name		Name	Last commit message	Last commit date
Latest commit History 4 Commits
configs		configs
data		data
src		src
README.md		README.md
input_pipeline.py		input_pipeline.py
models_cnn.py		models_cnn.py
models_graph_mixer.py		models_graph_mixer.py
models_graph_vit.py		models_graph_vit.py
train_graph_vit.py		train_graph_vit.py
train_mixer.py		train_mixer.py
training_loop.py		training_loop.py

Provide feedback