16 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	5fcd6ed221	Unit test fixes for when AF2 is not installed	2023-11-02 15:57:23 -04:00
Christina Floristean	82bda2d6d3	Refactored multimer config update	2023-08-03 20:52:06 -04:00
Christina Floristean	30764cf9e0	Minor fixes/reformatting for recent multimer training PR	2023-08-03 11:38:07 -04:00
Christina Floristean	68828c49c4	Multimer v3 updates	2023-04-17 14:25:45 -04: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	2faff4516b	Make template code use less memory during inference	2021-10-28 18:55:57 -04: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