Files

 main

/

tests

/

Copy path

Directory actions

More options

Directory actions

More options

Latest commit

aseyboldt

and

patrick-kidger

Allow zero shapes in nn.Linear

Feb 15, 2025

d08de05 · Feb 15, 2025

History

History

/

tests

/

Top

Folders and files

Name		Name	Last commit message	Last commit date
parent directory ..
__init__.py		__init__.py	Tidied helper into a relative import (#133)	Jul 12, 2022
conftest.py		conftest.py	Now using strict rank and dtype promotions when testing.	Dec 11, 2023
helpers.py		helpers.py	Switch to ruff-format and ruff for ipynb	Dec 10, 2023
requirements.txt		requirements.txt	Revert "Add [test] and update CONTRIBUTING (#848)"	Sep 14, 2024
test_abstract.py		test_abstract.py	updated ruff (#727)	May 18, 2024
test_ad.py		test_ad.py	Fix jacfwd (#734)	May 23, 2024
test_caches.py		test_caches.py	Switch to ruff-format and ruff for ipynb	Dec 10, 2023
test_callback.py		test_callback.py	Switch to ruff-format and ruff for ipynb	Dec 10, 2023
test_checkpoint.py		test_checkpoint.py	Switch to ruff-format and ruff for ipynb	Dec 10, 2023
test_closure_to_pytree.py		test_closure_to_pytree.py	Now using strict rank and dtype promotions when testing.	Dec 11, 2023
test_debug.py		test_debug.py	Support complex dtypes in networks (#765)	Jun 24, 2024
test_enum.py		test_enum.py	Switch to ruff-format and ruff for ipynb	Dec 10, 2023
test_errors.py		test_errors.py	Filtering out jax.pure_callback failures via logging; added test.	Sep 14, 2024
test_eval_shape.py		test_eval_shape.py	Switch to ruff-format and ruff for ipynb	Dec 10, 2023
test_filters.py		test_filters.py	Switch to ruff-format and ruff for ipynb	Dec 10, 2023
test_finalise_jaxpr.py		test_finalise_jaxpr.py	Upgrade jax and fix deprecation warnings	Dec 21, 2024
test_jit.py		test_jit.py	Fix AOT compilation for eqx.filter_jit with kwargs	Dec 22, 2023
test_make_jaxpr.py		test_make_jaxpr.py	Now using strict rank and dtype promotions when testing.	Dec 11, 2023
test_misc.py		test_misc.py	Switch to ruff-format and ruff for ipynb	Dec 10, 2023
test_module.py		test_module.py	Fix __init__ + __init_subclass__ + AbstractClassVar	Sep 18, 2024
test_nn.py		test_nn.py	Allow zero shapes in nn.Linear	Feb 15, 2025
test_noinline.py		test_noinline.py	Fixes for compatibility with JAX version 0.4.28.	May 12, 2024
test_nontraceable.py		test_nontraceable.py	More fixes for warnings raised in JAX 0.4.38.	Dec 21, 2024
test_onnx.py		test_onnx.py	Switch to ruff-format and ruff for ipynb	Dec 10, 2023
test_pformat.py		test_pformat.py	Now using wadler_lindig pprint library.	Jan 8, 2025
test_pmap.py		test_pmap.py	Remove if_mapped to skip a usage of find_top_trace.	Dec 8, 2024
test_primitive.py		test_primitive.py	Upgrade jax and fix deprecation warnings	Dec 21, 2024
test_scan.py		test_scan.py	Switch to ruff-format and ruff for ipynb	Dec 10, 2023
test_serialisation.py		test_serialisation.py	Deserialisation now works with filter_eval_shape. Closes #259.	Jan 10, 2024
test_sharding.py		test_sharding.py	Support complex dtypes in networks (#765)	Jun 24, 2024
test_shared.py		test_shared.py	Switch to ruff-format and ruff for ipynb	Dec 10, 2023
test_stateful.py		test_stateful.py	Switch to ruff-format and ruff for ipynb	Dec 10, 2023
test_str2jax.py		test_str2jax.py	Switch to ruff-format and ruff for ipynb	Dec 10, 2023
test_tree.py		test_tree.py	Make _DinstinctTuple inherit from tuple (#819)	Aug 29, 2024
test_update.py		test_update.py	Switch to ruff-format and ruff for ipynb	Dec 10, 2023
test_vmap.py		test_vmap.py	Remove if_mapped to skip a usage of find_top_trace.	Dec 8, 2024
test_while_loop.py		test_while_loop.py	Better AD for checkpointed while loop. (#697)	Apr 14, 2024
View all files