Expand file tree

/

jax

/

Copy path

Directory actions

More options

More options

Directory actions

More options

More options

Latest commit

 

History

History

History

/

jax

/

Copy path

Folders and files

Name		Name	Last commit message	Last commit date
parent directory ..
_src		_src
example_libraries		example_libraries
experimental		experimental
extend		extend
image		image
interpreters		interpreters
lax		lax
lib		lib
nn		nn
numpy		numpy
ops		ops
pallas		pallas
scipy		scipy
tools		tools
BUILD		BUILD
__init__.py		__init__.py
ad_checkpoint.py		ad_checkpoint.py
api_util.py		api_util.py
avals.py		avals.py
collect_profile.py		collect_profile.py
core.py		core.py
custom_batching.py		custom_batching.py
custom_derivatives.py		custom_derivatives.py
debug.py		debug.py
distributed.py		distributed.py
dlpack.py		dlpack.py
dtypes.py		dtypes.py
errors.py		errors.py
export.py		export.py
ffi.py		ffi.py
flatten_util.py		flatten_util.py
monitoring.py		monitoring.py
profiler.py		profiler.py
py.typed		py.typed
random.py		random.py
ref.py		ref.py
sharding.py		sharding.py
stages.py		stages.py
test_util.py		test_util.py
tree.py		tree.py
tree_util.py		tree_util.py
typing.py		typing.py
version.py		version.py