GitHub - sayakpaul/ViT-jax2tf: This repository hosts code for converting th...

This repository hosts code for converting the original Vision Transformer models (JAX) to TensorFlow...

0 users, 2 mentions 2021/09/21 16:38
JAX/Flaxを使ってMNISTを学習させてみる | TC3株式会社

ご存知の通り、TensorFlow、Keras、PyTorch(Chainer…)と近年は様々な深層学習ライブラリが使われています。最近、Jaxというライブラリが話題になっているものの、十分すぎるライ...

20 users, 6 mentions 2022/02/09 23:00
Jax/Flax × TransformersでBERTのfine-tuningをTPUで行う | 株式会社AI Shift

AI ShiftのTECH BLOGです。AI技術の情報や活用方法などをご案内いたします。

1 users, 2 mentions 2021/10/26 12:10
GitHub - kingoflolz/mesh-transformer-jax: Model parallel transformers in JA...

Model parallel transformers in JAX and Haiku. Contribute to kingoflolz/mesh-transformer-jax developm...

3 users, 35 mentions 2021/07/04 06:08
JAXによるスケーラブルな機械学習 - ZOZO TECH BLOG

はじめに こんにちは、ZOZO NEXT ZOZO ResearchのSai Htaung Khamです。ZOZO NEXTは、ファッション領域におけるユーザーの課題を想像しテクノロジーの力で解決する...

36 users, 13 mentions 2022/03/17 02:00
PyTorch to JAX 移行ガイド(GPUでのCNN学習 | BatchNorm編)

Python deeplearning PyTorch # jax tech 背景 前回: PyTorch to JAX 移行ガイド(MLP学習編) JAXベースのNNライブラリであるFl...

1 users, 3 mentions 2021/10/02 15:09
拡散生成モデルで学ぶJax/Flaxによる深層学習プログラミング

Python deeplearning MachineLearning # jax tech はじめに 深層学習モデルやその学習を実装する際には、多くの場合でPyTorchやKerasなどの...

3 users, 9 mentions 2022/07/31 01:37
NumPyroとJax Numpyで時系列 - HELLO CYBERNETICS

はじめに コード モジュール データ モデル 事前分布からのサンプリング 推論 結果

1 users, 2 mentions 2020/05/07 14:53
JAX Course - 2. Working with Neural Networks in JAX - YouTube

📍 Q&A and forum discussion: wandb.me/JAX-2In this meetup, Sanyam will walk us through Neural Networ...

0 users, 0 mentions 2021/12/04 16:37
PyTorch to JAX 移行ガイド(MLP学習編)

Python PyTorch # jax tech 背景 「JAX最高」「GoogleではみんなJAXやってる」などと巷で言われているが、研の活動をやってると、比較手法がPyTorchで提供さ...

7 users, 5 mentions 2021/09/27 09:08