2024-07-08

Added

• Dataset generation script now in data_gen/: (a) gns_data moved in there and (b) lagrangebench_data was added #30
• Better assertions and comments #31
• Note on missing determinism of GPU operations - see README.md #31

Fixed

• Docs #30
• model.init inputs are now also JAX arrays #31
• Fix working with external datasets #31

Changed

• Update JAX to latest 0.4.29 #30
• Remove JAX-MD from dependencies. Now neighbor search happens through JAX-SPH #30
• [BREAKING] cfg.dataset_path became cfg.dataset.src #31

---

Full Changelog: v0.2.0a1...v0.2.0