4 Star 10 Fork 2

Gitee 极速下载/JAX

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
.github
benchmarks
build
ci
cloud_tpu_colabs
docs
_static
_templates
_tutorials
debugging
export
ffi
jep
notebooks
pallas
sphinxext
README.md
about.md
advanced-autodiff.md
advanced_guide.rst
aot.md
api_compatibility.md
async_dispatch.rst
autodidax.ipynb
autodidax.md
autodidax.py
autodidax2_part1.ipynb
autodidax2_part1.md
autodidax2_part1.py
automatic-differentiation.md
automatic-vectorization.md
beginner_guide.rst
build_custom_gpu.sh
building_on_jax.md
changelog.md
concurrency.rst
conf.py
config_options.rst
contributing.md
contributor_guide.rst
control-flow.md
debugging.md
default_dtypes.md
deprecation.md
developer.md
device_memory_profiling.md
distributed_data_loading.md
errors.rst
extensions.rst
external-callbacks.md
faq.rst
ffi.ipynb
ffi.md
glossary.rst
gpu_memory_allocation.rst
gpu_performance_tips.md
gradient-checkpointing.md
hero.html
index.rst
installation.md
investigating_a_regression.md
jax-primitives.md
jax.debug.rst
jax.distributed.rst
jax.dlpack.rst
jax.dtypes.rst
jax.example_libraries.optimizers.rst
jax.example_libraries.rst
jax.example_libraries.stax.rst
jax.experimental.checkify.rst
jax.experimental.compilation_cache.rst
jax.experimental.custom_dce.rst
jax.experimental.custom_partitioning.rst
jax.experimental.jet.rst
jax.experimental.key_reuse.rst
jax.experimental.mesh_utils.rst
jax.experimental.multihost_utils.rst
jax.experimental.pallas.mosaic_gpu.rst
jax.experimental.pallas.rst
jax.experimental.pallas.tpu.rst
jax.experimental.pallas.triton.rst
jax.experimental.pjit.rst
jax.experimental.rst
jax.experimental.serialize_executable.rst
jax.experimental.shard_map.rst
jax.experimental.sparse.rst
jax.export.rst
jax.extend.core.rst
jax.extend.linear_util.rst
jax.extend.mlir.rst
jax.extend.random.rst
jax.extend.rst
jax.ffi.rst
jax.flatten_util.rst
jax.image.rst
jax.lax.rst
jax.lib.rst
jax.nn.initializers.rst
jax.nn.rst
jax.numpy.rst
jax.ops.rst
jax.profiler.rst
jax.random.rst
jax.rst
jax.scipy.rst
jax.sharding.rst
jax.stages.rst
jax.test_util.rst
jax.tree.rst
jax.tree_util.rst
jax.typing.rst
jax_array_migration.md
jaxpr.md
jit-compilation.md
key-concepts.md
multi_process.md
notes.rst
persistent_compilation_cache.md
profiling.md
pytrees.md
quickstart.md
random-numbers.md
rank_promotion_warning.rst
requirements.txt
sharded-computation.ipynb
sharded-computation.md
stateful-computations.md
transfer_guard.rst
tutorials.rst
type_promotion.rst
user_guides.rst
working-with-pytrees.md
xla_flags.md
examples
images
jax
jax_plugins
jaxlib
tests
third_party
.bazelrc
.bazelversion
.editorconfig
.gitignore
.pre-commit-config.yaml
.readthedocs.yml
AUTHORS
BUILD.bazel
CHANGELOG.md
CITATION.bib
CONTRIBUTING.md
LICENSE
README.md
WORKSPACE
build_wheel.py
conftest.py
platform_mappings
pyproject.toml
setup.py
此仓库是为了提升国内下载速度的镜像仓库,每日同步一次。 原始仓库: https://github.com/google/JAX
克隆/下载
贡献代码
同步代码
取消
提示: 由于 Git 不支持空文件夾,创建文件夹后会生成空的 .keep 文件
Loading...
README

To rebuild the documentation, see Update Documentation.

马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/mirrors/JAX.git
git@gitee.com:mirrors/JAX.git
mirrors
JAX
JAX
main

搜索帮助