jaxnet

Concise deep learning for JAX

APACHE-2.0 License

Downloads
568
Stars
184
View Code on GitHub
Ecosystems: Python

Issue Statistics

Past Year

All Time

Total Pull Requests
0
1
Merged Pull Requests
0
1
Total Issues
0
25
Time to Close Issues
N/A
22 days
Package Rankings
Top 15.07% on Pypi.org
Badges
Extracted from project README
PyPI
Related Projects
tensorf-jax

Unofficial implementation of Tensorial Radiance Fields (Chen & Xu ‘22)

17 Apr 2022 38
sbijax

Simulation-based inference in JAX

12 Oct 2022 12
fidjax

Frechet Inception Distance in JAX

15 Jul 2024 1
keras

Deep Learning for humans

28 Mar 2015 60,873
torch2jax

Run PyTorch in JAX. 🤝

03 Oct 2023 192
jraph

A Graph Neural Network Library in Jax

23 Nov 2020 1,327
equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

29 Jul 2021 1,846
lpips-j

Minimal JAX/Flax port of `lpips` supporting `vgg16`, with pre-trained weights stored in the 🤗 Hug...

08 Jul 2022 14
efficientnet-jax

EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax

12 Sep 2020 127
jaxtyping

Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https...

23 Jun 2022 1,131
dm-haiku

JAX-based neural network library

18 Feb 2020 2,816
ninjax

General Modules for JAX

23 May 2022 57
lineax

Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax

09 May 2023 345
JAXSeq

Train very large language models in Jax.

02 Oct 2022 193
parallax

19 May 2020 157