Comments (3)
Looks like jacfwd
and jacrev
had some transpose issues as well!
from jax.
Please reopen if you run into more issues with these particular primitives, or open new issues as you need vmap support for other primitives!
from jax.
Thanks for fixing this so quickly!
from jax.
Related Issues (20)
- jax2tf oddly casts uint8 input to uint32 before passing to radix sort. HOT 2
- Unexpected behavior with jax.lax.conv_general_dilated HOT 2
- PRNG Key in shard_map() HOT 3
- JIT Compilation Error with Single Vmap on GPU HOT 1
- Why is `jax.tree_util.equality_errors` not exported? HOT 1
- Numerical precision differences in jitted vs. non-jitted code! HOT 3
- JAX indexing should support zero-dimensional boolean masks
- Memory profiler missing in trace HOT 6
- Getting some problems in the usage of jnp.linalg.inv HOT 3
- Installing jaxlib with CUDA Version: 11.7 HOT 5
- AssertionError raised due to failing assertion in partial_eval.py HOT 4
- Extending JAX with CUDA: cstdint: error: the global scope has no "..." HOT 3
- Complex gradient HOT 3
- Reconstructing a custom registered dataclass seems to give a segmentation fault
- AttributeError: module 'jax.random' has no attribute 'KeyArray' HOT 5
- Accessing incoming gradient and activations in custom_vjp HOT 4
- CUDA `XlaRuntimeError` with MPI on `jax==0.4.31` HOT 5
- Finite precision error calculations always 0 under JIT with bfloat16 HOT 3
- Errors from interaction between `shard_map` with `auto` and `lax.map`
- Document Named Collectives HOT 1
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.