Basant1861
Basant1861
The output of compute_pg is a (12 ,1) 2D array . This is a minimal reproducible example. ``` import jax.numpy as jnp import numpy as np from functools import partial...
I am using the following code snippet (specifically the function **load_torch_weights**) but it uses some equinox methods(**eqx.experimental.set_state** and **eqx.experimental.StateIndex**) which seem no longer supported in the latest version of equinox....
Hello, Is it possible to run equinox on jetson xavier? if yes, could you please point me towards relevant resources? We basically want to run equinox onboard a jetson xavier...