BobMcDear/flaim
Flax Image Models - State-of-the-art pre-trained vision backbones for Flax.
This is a tool for machine learning engineers and researchers who are building computer vision applications. It provides a collection of pre-trained image models that can be used as building blocks for tasks like image classification, object detection, or image generation. You input raw image data, and it outputs model predictions or features, helping you accelerate development and leverage state-of-the-art architectures.
No commits in the last 6 months. Available on PyPI.
Use this if you need to quickly integrate high-performance, pre-trained image recognition capabilities into your Flax-based machine learning project.
Not ideal if you are looking for a no-code solution or a tool for general data science tasks outside of deep learning computer vision.
Stars
23
Forks
1
Language
Python
License
GPL-3.0
Category
Last pushed
Jun 05, 2025
Commits (30d)
0
Dependencies
4
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/BobMcDear/flaim"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
explosion/thinc
🔮 A refreshing functional take on deep learning, compatible with your favorite libraries
google-deepmind/optax
Optax is a gradient processing and optimization library for JAX.
patrick-kidger/diffrax
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable....
google/grain
Library for reading and processing ML training data.
patrick-kidger/equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/