fusions

Dependencies (8)

• anesthetic <3.0.0,>=2.4.2
• diffrax <0.5.0,>=0.4.0
• distrax <0.2.0,>=0.1.5
• flax <0.9.0,>=0.8.2
• optax <0.3.0,>=0.2.2
• pytest <8.0,>=7.3
• torch <3.0.0,>=2.1.0
• tqdm <5.0.0,>=4.62.0