Carlos Andres del Valle

readdlm not working with white spaces

7

readdlm ignores all-white spaces by default when a delimiter is not specified. However, when one wants to specify the data type to be read it is obligatory to specify the...

CapsLock not working: Linux

3

### Check for existing issues - [X] Completed ### Describe the bug / provide steps to reproduce it I installed Zed today (Jul 8) following the instructions for Linux: rustup...

How to compute gradient of function that mixes loops

1

Hi, I'm currently triying to compute gradient of a potential energy definition for a mesh. The idea is to use this gradient to calculate forces in the vertices. The problem...

Fix small typo in performance consideration docs

2

Hi Flax team, Here is a fix to a minor typo in the documentation of performance considerations for nnx.

How to train an NNX netowork using scan and jit?

8

Hi Flax team, I'm writing as I've been struggling the past week trying to implement PPO for reinforcement learning using NNX. I'm at the point where I need some help...

Feature/Docs request: Saving and Loading checkpoints with optimizers and metrics

2

Dear Flax team, I've been setting up all the infrastructure to save and load my flax models using Orbax and the guide in the [documentation](https://flax.readthedocs.io/en/latest/guides/checkpointing.html). However, I noticed that a...

Hi, I'm using Distrax distributions with some custom bijectors. This is an example of my bijector: ```python class MaxNormSpace(distrax.Bijector): r""" **Radial max-norm** constraint for vector actions: Scales the radius with...