基于JAX的神经网络库-Python开发资源-CSDN文库

共217个文件

py：127个

json：43个

build：10个

需积分: 12 137 浏览量 2021-05-25 15:27:14 上传评论收藏 556KB ZIP 举报

资源推荐

资源详情

资源评论

收起资源包目录

package

基于JAX的神经网络库-Python开发（217个子文件）

references.bib 4KB

BUILD 14KB

BUILD 5KB

BUILD 2KB

BUILD 2KB

BUILD 2KB

BUILD 1KB

BUILD 736B

BUILD 659B

BUILD 555B

BUILD 244B

build_defs.bzl 2KB

.gitignore 2KB

.gitignore 7B

MANIFEST.in 141B

non_trainable.ipynb 203KB

jax2tf.ipynb 56KB

visualization.ipynb 46KB

build_your_own_haiku.ipynb 42KB

vqvae_example.ipynb 20KB

basics.ipynb 19KB

mnist_gan.ipynb 18KB

haiku_lstms.ipynb 18KB

transforms.ipynb 1KB

nets.ResNet.json 30KB

nets.MobileNetV1.json 4KB

nets.VectorQuantizerEMA.json 608B

BatchNorm.json 601B

MultiHeadAttention.json 551B

Recurrent-Conv3DLSTM-static_unroll.json 392B

Recurrent-Conv3DLSTM-dynamic_unroll.json 392B

SeparableDepthwiseConv2D.json 389B

Recurrent-Conv2DLSTM-dynamic_unroll.json 379B

Recurrent-Conv2DLSTM-static_unroll.json 379B

Recurrent-Conv1DLSTM-dynamic_unroll.json 366B

Recurrent-Conv1DLSTM-static_unroll.json 366B

nets.MLP.json 365B

Recurrent-VanillaRNN-static_unroll.json 305B

Recurrent-VanillaRNN-dynamic_unroll.json 305B

ConvNDTranspose.json 266B

nets.VectorQuantizer.json 263B

Conv3DTranspose.json 260B

DepthwiseConv2D.json 255B

Conv2DTranspose.json 253B

LayerNorm.json 247B

ConvND.json 247B

Conv1DTranspose.json 246B

InstanceNorm.json 245B

Conv3D.json 241B

Conv2D.json 234B

Recurrent-GRU-static_unroll.json 229B

Recurrent-GRU-dynamic_unroll.json 229B

Conv1D.json 227B

GroupNorm.json 217B

Recurrent-LSTM-static_unroll.json 210B

Recurrent-LSTM-dynamic_unroll.json 210B

SpectralNorm.json 205B

Linear.json 203B

Embed.json 198B

RMSNorm.json 177B

Bias.json 172B

Recurrent-ResetCore-static_unroll.json 82B

Recurrent-ResetCore-dynamic_unroll.json 82B

Sequential.json 77B

Recurrent-IdentityCore-dynamic_unroll.json 69B

Recurrent-IdentityCore-static_unroll.json 69B

Flatten.json 68B

LICENSE 11KB

Makefile 581B

README.md 16KB

README.md 2KB

CONTRIBUTING.md 2KB

README.md 1KB

README.md 924B

README.md 524B

conv.py 31KB

conv_test.py 28KB

base.py 27KB

recurrent.py 26KB

module_test.py 26KB

module.py 26KB

stateful.py 23KB

stateful_test.py 22KB

resnet.py 18KB

transform_test.py 17KB

base_test.py 17KB

recurrent_test.py 16KB

summarise.py 15KB

transform.py 14KB

vqvae.py 14KB

train.py 13KB

dot.py 13KB

data_structures_test.py 12KB

basic.py 11KB

mnist_pruning.py 11KB

dataset.py 11KB

filtering_test.py 11KB

initializers.py 11KB

group_norm_test.py 10KB

depthwise_conv_test.py 10KB

共 217 条

内容反馈

彷徨的牛

粉丝: 58
资源: 4720

最新资源

资源上传下载、课程学习等过程中有任何疑问或建议，欢迎提出宝贵意见哦~我们会及时处理！点击此处反馈

feedback-tip