deps: add dep groups for dask and jax#275

lucascolley merged 1 commit intodata-apis:maindata-apis/array-api-extra:mainfrom

lucascolley:deps-groupslucascolley/array-api-extra:deps-groupsCopy head branch name to clipboard