Comments (1)
Hi - thanks for the report! The Rotation
functionality has some implementation issues, and is a part of the package that we've identified (retroactively) as out-of-scope for JAX (see https://jax.readthedocs.io/en/latest/jep/18137-numpy-scipy-scope.html#scipy-spatial), and at some point in the future it will probably be removed.
My hope is that ongoing efforts to make scipy compatible with the Python array API will allow JAX users to replace these tools with using the scipy rotation code directly, although that's not yet possible.
In the meantime, is this an issue that you can work around?
from jax.
Related Issues (20)
- CuDNN Initialization Error with Pre-existing CUDA Environment Fails with HOT 5
- Pallas TPU reduce kernel generalization gives wrong numerical results? HOT 3
- Passing `DeviceLocalLayout` to `jax.jit` causes incorrect outputs on GPU
- Precision of jnp.linalg.solve HOT 3
- JAX segment_sum is two times slower for FP16 inputs than FP32 inputs HOT 10
- jax library error jaxlib.xla_extension.XlaRuntimeError: FAILED_PRECONDITION: DNN library initialization failed HOT 3
- arctan2 and atan2 argument names are misleading HOT 3
- Jax Einsum Implementation Error HOT 1
- jaxlib.xla_extension.XlaRuntimeError: FAILED_PRECONDITION: DNN library initialization failed. HOT 2
- Improve support for using JAX's custom LAPACK calls externally HOT 2
- `erf_inv` not supported in Pallas GPU
- Unsupported conversion from f64 to f16 in pallas despite not using fp16 HOT 1
- Pallas kernel crash at `llo::CouldLtS32` when `interpret=False` on TPU HOT 3
- Pallas Unexpected Numeric Results With `tf32` on A100 GPU HOT 1
- [Pallas GPU] `jnp.abs()` gives the wrong result HOT 1
- Setting `JAX_PLATFORMS=cpu` has no effect. HOT 6
- Guard against unintentional transfers to CPU HOT 2
- JITing loss function causes erroneous results
- `jax.nn.dot_product_attention(...implementation='cudnn')` throws error `libnvrtc.so.12` not found
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from jax.