-
## Project description
At the moment there are various flavors for compilers and stdenvs in `llvmPackages`. I believe most of the existing configurations are not very useful and the most important …
-
## Issue description
Trying computing the model jacobian using a keras model returns zeros
## Code example
```python
import keras
import torch
model = keras.Sequential(
[keras.layers.De…
-
Dear team,
I'm having an issue restoring pure dict from nnx.state,
Orbax will change the key has int type into string type.
Is there any way to fix it?
I put a sample code that can reproduce the issu…
-
# Motivation: Make JAX relevant to many more users
JAX provides a powerful general-purpose tool for automatic differentiation, but it usually requires that users write code that is JAX-traceable en…
-
### Review Mojo's priorities
- [X] I have read the [roadmap and priorities](https://docs.modular.com/mojo/roadmap.html#overall-priorities) and I believe this request falls within the priorities.
###…
-
i have a couple of links but I'm slightly confused what the recommended jax implementation of flash-attention.
here are the links:
- jupyter: https://github.com/jax-ml/jax-triton/blob/main/exampl…
-
I need this for finding the eigenvectors of the Hessian after tridiagonalizing it with Lanczos iteration. Right now the function looks like:
```python
def eigh_tridiagonal(d: ArrayLike, e: ArrayLi…
-
This package is very similar to JAX's provided, and improving, [external callback functionality](https://jax.readthedocs.io/en/latest/notebooks/external_callbacks.html).
I also urge you to show sup…
-
I noticed earlier today that Haiku has SpectralNormalization -- very cool!
I'm interested in implementing an improved version, which does a much better job estimating the norm for convolutional lay…
-
Here's an overview of the features we intend to work on in the near future.
## Core Keras
### Saving & export
- Implement saving support for sharded models (sharded weights files).
- Improve…