14 Commits

Author	SHA1	Message	Date
Jennifer	30195c4abb	Adds absolute error comparison function with better messaging.	2024-02-01 22:53:12 -05:00
Jennifer	8f8b537da8	update deprecated jax.numpy.DeviceArray to jax.Array	2024-02-01 22:53:12 -05:00
Christina Floristean	39a6d0e64a	Merging in main branch	2023-04-09 11:07:04 -04:00
Christina Floristean	d8ee9c5f34	All non-cuda tests passing for monomer/multimer. Tri mul/attn and OPM order switched.	2023-02-17 00:59:28 -05:00
Gustaf Ahdritz	700fe8fee2	Remove print statements	2023-02-04 00:29:20 -06:00
Gustaf Ahdritz	e14a313eba	Fix test bugs	2023-02-04 00:22:28 -06:00
Gustaf Ahdritz	6ce8cfe31e	Fixes	2022-02-03 17:06:45 -05:00
Gustaf Ahdritz	6c5cd945bd	Make triangular attention test even more sensitive	2021-12-10 23:14:24 -05:00
Gustaf Ahdritz	80f968fe50	Make triangle attention test more sensitive	2021-12-10 20:48:59 -05:00
Gustaf Ahdritz	497670998b	Bring tests up to speed	2021-10-19 23:07:09 -04:00
Gustaf Ahdritz	07e642671c	Standardize code style	2021-10-16 01:17:18 -04:00
Gustaf Ahdritz	9f6b67f37c	Fix another FP16 overflow, finish tests, update util scripts	2021-10-06 18:53:45 -04:00
Gustaf Ahdritz	304b5ff7cb	Begin to spruce up unit tests, fix config	2021-10-02 02:52:44 -04:00
Gustaf Ahdritz	92f1932e6b	Initial commit	2021-09-17 18:34:59 -04:00