12 commits

This branch

This branch

All branches

Author	SHA1	Message	Date
Lukas Winkler	a80cbafe1a	add typing	2024-08-02 17:02:39 +02:00
Lukas Winkler	4381a7998e	change output format slightly and add shard_map example	2024-07-29 15:29:03 +02:00
Lukas Winkler	9ac3ea6715	document print_array_stats and add dtype to it	2024-06-21 12:18:31 +02:00
Lukas Winkler	e1dc02aee2	Merge branch 'main' of github.com:Findus23/jax-array-info	2023-10-25 13:38:47 +02:00
Lukas Winkler	fd64100565	adapt tests for results with jax 0.4.19	2023-10-25 13:37:36 +02:00
Lukas Winkler	ad1c2d5e91	add stats over all currently allocated arrays	2023-10-09 17:15:23 +02:00
Lukas Winkler	eaf968a9c6	one more test	2023-09-04 16:47:53 +02:00
Lukas Winkler	f75389aa0a	one more jit example	2023-09-04 16:42:32 +02:00
Lukas Winkler	7cf9747b55	minor fixes	2023-09-04 16:33:38 +02:00
Lukas Winkler	1a591faf79	add installation	2023-09-04 16:26:03 +02:00
Lukas Winkler	169e6a3fe6	add tests	2023-09-04 16:22:58 +02:00
Lukas Winkler	58694cb800	initial version	2023-09-04 15:34:05 +02:00