Build and visualize neural networks in JAX
This project is no longer maintained, so it won't get fixes or updates.
- Skill level
- For developersA building block for programmers. You'll write some code to use it.
- How you use it
- Code library
- Cost & license
- Free, even for businessApache-2.0
- Stars & last update
- ★ 1,899No longer maintained · 22 Jun 2025
- Health
- Use with careThe owner archived it, so no more fixes · Clear open-source license · Around for over a year
- Try before installing
- No official online demo
Setup, requirements and trust checks
A plain-language explanation, what you need, how to start, the trust checks, a fits-my-device verdict and the project's own instructions.
Join Verbito to see setup steps, trust checks and guides
The full Atlas is included in Verbito Starter and above, and in the free 7-day Pro trial for new accounts.
The full Atlas is included in Starter and above
Upgrade to see setup steps, trust checks, walkthroughs and guides, and to vote and share tips.
Did this work for you?
Loading results…
Similar tools
Use hundreds of pretrained image models in PyTorch
pytorch-image-models by huggingface
Build and train machine-learning models (for developers)tensorflow by tensorflow
Build and train neural networks (for developers)pytorch by pytorch
Build classic machine-learning models in Pythonscikit-learn by scikit-learn
fine-tuninginterpretabilityjaxneural-networksvisualization