JAX: JAX 是一个 TensorFlow 的简化库，它结合了 Autograd 和 XLA，专门用于高性能机器学习研究

Gitee Talk | 模力方舟 AI 应用开发沙龙第六站 · 8月23日广州集结中！

加入 Gitee

与超过 1200万开发者一起发现、参与优秀开源项目，私有仓库也完全免费：）

免费加入

文件

.github

benchmarks

build

cloud_tpu_colabs

docs

examples

images

jax

_src

example_libraries

experimental

extend

image

__init__.py

interpreters

lax

lib

numpy

ops

scipy

tools

BUILD

__init__.py

ad_checkpoint.py

api_util.py

cloud_tpu_init.py

collect_profile.py

core.py

custom_batching.py

custom_derivatives.py

custom_transpose.py

debug.py

distributed.py

dlpack.py

dtypes.py

errors.py

export.py

ffi.py

flatten_util.py

monitoring.py

profiler.py

py.typed

random.py

ref.py

sharding.py

stages.py

test_util.py

tree.py

tree_util.py

typing.py

util.py

version.py

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

test_shard_count.bzl

此仓库是为了提升国内下载速度的镜像仓库，每日同步一次。原始仓库： https://github.com/google/JAX

克隆/下载

Update references to the GitHub url in JAX... d4d1518 11个月前

29416 次提交

...

__init__.py

马建仓 AI 助手

尝试更多

代码解读

代码找茬

代码优化

Python

https://gitee.com/mirrors/JAX.git

git@gitee.com:mirrors/JAX.git

mirrors

JAX

main

Gitee 极速下载/JAX

简介

发行版

贡献者

语言

近期动态

Gitee 极速下载/JAX .gitee-modal { width: 500px !important; }

简介

发行版

贡献者

语言

近期动态

搜索帮助

Gitee 极速下载/JAX