captum
flax
Our great sponsors
captum | flax | |
---|---|---|
11 | 10 | |
4,568 | 5,520 | |
2.5% | 5.1% | |
8.6 | 9.7 | |
3 days ago | about 22 hours ago | |
Python | Python | |
BSD 3-clause "New" or "Revised" License | Apache License 2.0 |
Stars - the number of stars that a project has on GitHub. Growth - month over month growth in stars.
Activity is a relative number indicating how actively a project is being developed. Recent commits have higher weight than older ones.
For example, an activity of 9.0 indicates that a project is amongst the top 10% of the most actively developed projects that we are tracking.
captum
-
[D] [R] Research Problem about Weakly Supervised Learning for CT Image Semantic Segmentation
Most likely, NNs in general love shortcut learning (see Geirhos et al. 2020). In general, local explanations such as grad-cam are quite noisy, and sometimes even inconsistent (see Seo et al. 2018 ). Now, in my experience, I've seen that integrated gradients (see Sundararajan et al 2017) does a better job compared to Grad-CAM (also, add a noise tunnel), but this is only based on my limited experience. I would totally recommend using the implementations from the Captum library for loca explanations.
-
[D] Off-the-shelf image saliency scoring models?
Take a look at captum
-
Can you interrogate a machine learning model to find out why it gave certain predictions?
Sometimes. If explainable predictions are part of your business requirements, it's probably better not to rely entirely on black box models and instead design a system that gives you the information you need as part of it. If you end up using black box models, there are still methods that attempt to help attribute explanations to your prediction. Here's an example of a toolkit for attributing explanations post-hoc to black box model predictions: https://github.com/pytorch/captum
- [D] DL Practitioners, Do You Use Layer Visualization Tools s.a GradCam in Your Process?
-
What kind of explainability techniques exist for Reinforcement learning?
The straightforward way to interpret RL agent's decision is to use captum library.
-
[D] How do you choose which Black-Box Explainability method to use?
My use-case is research-oriented. I work on Explainable AI. Generally, the best package I've come across to compute attributions in Pytorch Captum. If your object detector is in PyTorch, you can perhaps build it in.
-
PyTorch vs. TensorFlow in 2022
Do any JAX experts know if there is an equivalent to https://captum.ai/ - a model interpretability library for pytorch?
In particular i want to be able to measure feature importance on both inputs and internal layers on a sample by sample basis. This is the only thing currently holding me back from using JAX right now.
Alternatively a simle to read/understand/port implementation of DeepLIFT would work too.
thanks
-
DeepLIFT or other explainable api implementations for JAX (like captum for pytorch)?
I'm interested to use JAX but am having a hard time finding anything similar to captum for the pytorch world.
- how to extract features from a (CNN) convolutional network having raw data with (XAI) explainable techinques?
-
Looking for help regarding explainable AI
Do you actually want to implement something? There are decent explainability libraries now, e.g., [AIX360](https://aix360.mybluemix.net/), [InterpretML](https://interpret.ml/), or [captum](https://captum.ai/). Pytorch + maybe pytorch lightning + captum might be the quickest way to actually implement something like an explainable neural net yourself. Do the standard tutorials for each of them and watch a few YT videos (or follow a coursera course or something like that) about how these things work in theory and practice, and you'll get up to speed relatively quickly. You will not be able to do useful work in ML without actually learning the ropes.
flax
-
Maxtext: A simple, performant and scalable Jax LLM
Is t5x an encoder/decoder architecture?
Some more general options.
The Flax ecosystem
https://github.com/google/flax?tab=readme-ov-file
or dm-haiku
https://github.com/google-deepmind/dm-haiku
were some of the best developed communities in the Jax AI field
Perhaps the “trax” repo? https://github.com/google/trax
Some HF examples https://github.com/huggingface/transformers/tree/main/exampl...
Sadly it seems much of the work is proprietary these days, but one example could be Grok-1, if you customize the details. https://github.com/xai-org/grok-1/blob/main/run.py
-
What is the JAX/Flax equivalent of torch.nn.Parameter?
https://github.com/google/flax/discussions/919 https://flax.readthedocs.io/en/latest/_modules/flax/linen/attention.html
-
Announcing flax 0.2 - A fully featured ECS
Just as an FYI, you might be competing against another big open source project with the same name https://github.com/google/flax
-
Flax: How to use one linen module inside another for training?
I have asked the same question on the Flax discussion page on Github as well.
-
[D] Should We Be Using JAX in 2022?
What's your favorite Deep Learning API for JAX - Flax, Haiku, Elegy, something else?
-
PyTorch vs. TensorFlow in 2022
As a researcher in RL & ML in a big industry lab, I would say most of my colleagues are moving to JAX 0https://github.com/google/jax], which this article kind of ignores. JAX is XLA-accelerated NumPy, it's cool beyond just machine learning, but only provides low-level linear algebra abstractions. However you can put something like Haiku [https://github.com/deepmind/dm-haiku] or Flax [https://github.com/google/flax] on top of it and get what the cool kids are using :)
- [D] Getting Started with Deep Learning in JAX with Treex in 16 lines
-
[D] JAX learning resources?
- https://github.com/google/flax/tree/main/examples
- Why would I want to develop yet another deep learning framework?
-
[D] Why is tensorflow so hated on and pytorch is the cool kids framework?
Any thoughts on Flax?
What are some alternatives?
shap - A game theoretic approach to explain the output of any machine learning model.
dm-haiku - JAX-based neural network library
DALEX - moDel Agnostic Language for Exploration and eXplanation
Pytorch - Tensors and Dynamic neural networks in Python with strong GPU acceleration
lucid - A collection of infrastructure and tools for research in neural network interpretability.
trax - Trax — Deep Learning with Clear Code and Speed
equinox - Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
WeightWatcher - The WeightWatcher tool for predicting the accuracy of Deep Neural Networks
Flux.jl - Relax! Flux is the ML library that doesn't make you tensor
alibi - Algorithms for explaining machine learning models
objax