4 Commits

Author	SHA1	Message	Date
Brett Kuprel	34df2b97df	previous commit broke colab example, so adjusting flax requirement to 0.4.2 for now	2022-06-28 08:04:08 -04:00
Brett Kuprel	38ebe54a38	works with latest flax version 0.5.2 now	2022-06-28 07:12:29 -04:00
Brett Kuprel	18e6a9852f	license and cleanup	2022-06-27 14:34:10 -04:00
Brett Kuprel	018414a5c3	fixed relative imports	2022-06-27 12:43:47 -04:00