jax[cpu]
e3nn_jax @ git+https://github.com/e3nn/e3nn-jax
ipykernel
matplotlib
lovelyplots
plotly
nbformat>=4.2.0
pandas
seaborn
tqdm
mdtraj