Comments (2)
Thanks a lot for the report! Should be fixed as of e6cc345.
from jax-md.
Oh also later in the notebook the second to last cell gives an error
---------------------------------------------------------------------------
TypeError Traceback (most recent call last)
<ipython-input-25-43e0542c4175> in <module>()
17 # Initialize the simulation
18
---> 19 state = init_fn(key, test_positions[0], Si_mass)
20 nbrs = neighbor_fn(state.position)
21
12 frames
/usr/local/lib/python3.6/dist-packages/haiku/_src/transform.py in apply_fn(params, state, rng, *args, **kwargs)
295 rng, err_msg=(APPLY_RNG_STATE_ERROR if state else APPLY_RNG_ERROR))
296 with base.new_context(params=params, state=state, rng=rng) as ctx:
--> 297 out = f(*args, **kwargs)
298 return out, ctx.collect_state()
299
TypeError: model() missing 1 required positional argument: 'neighbor'
which is fixed by adding a neighbor parameter state = init_fn(key, test_positions[0], Si_mass, neighbor=neighbor)
.
from jax-md.
Related Issues (20)
- Cannot import 'FunctionalFullyConnectedTensorProduct' from 'e3nn_jax' HOT 1
- test_nequip_silicon in energy_test.py is broken.
- test_nve_2d_neighbor_list_multi_atom_species in rigid_body_test.py is broken HOT 1
- documentation not compiling 2
- Elasticity calculations involving rigid bodies
- Out Of Memory issue during neighbor list generation
- Inconsistence of periodic and periodic_general
- equivariant_neural_networks notebook is broken
- Question about npt simulation HOT 2
- AttributeError: module 'jax.random' has no attribute 'KeyArray' HOT 1
- Neighbor lists are broken for rigid bodies.
- PME Energies
- AttributeError: module 'jax' has no attribute 'linear_util'
- ImportError in the example notebooks HOT 1
- Run sample notebooks as of 13 April 2024. HOT 2
- Potential Definition Discrepancy with Stress Calculation
- future directions to improve jax-md's performance?
- Error when running flocking.ipynb HOT 1
- How to generate equivalent NequIP model in JAX-MD
- test_coulomb_cubeions fails
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-md.