Update README.md
Browse files
README.md
CHANGED
|
@@ -6,5 +6,13 @@ colorTo: gray
|
|
| 6 |
sdk: static
|
| 7 |
pinned: false
|
| 8 |
---
|
|
|
|
|
|
|
|
|
|
| 9 |
|
| 10 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 6 |
sdk: static
|
| 7 |
pinned: false
|
| 8 |
---
|
| 9 |
+
<p align="center">
|
| 10 |
+
<image src="https://cdn-uploads.huggingface.co/production/uploads/6262be8108481d1d75edaad0/KeNFkl5dB4R2MMbH7IUhE.png" width="500"/>
|
| 11 |
+
</p>
|
| 12 |
|
| 13 |
+
### JaxNN: Foundation Models in JAX/Flax
|
| 14 |
+
|
| 15 |
+
**Jax** **N**eural **N**etworks is an open-source library for foundation models in JAX and Flax. It provides a unified framework for loading, creating, and using pretrained models (e.g., ResNet, ViT).
|
| 16 |
+
|
| 17 |
+
- [HuggingFace models](https://huggingface.co/models?library=jaxnn&sort=trending)
|
| 18 |
+
- [Github repository](https://github.com/Xrenya/jaxnn)
|