Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add 'jaxgym' package and ErgoCub environment #5

Closed
wants to merge 27 commits into from
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
27 commits
Select commit Hold shift + click to select a range
f1fbe68
Revert "Perform a deep copy when copying pytrees"
diegoferigo Jun 30, 2023
825ff5a
Always restore mutability before exiting the context manager
diegoferigo Jul 27, 2023
719eb40
Convert utils.py module to jaxsim.utils package
diegoferigo Jul 27, 2023
68ac4b5
Create abstract class that can be automatically vectorized with vmap
diegoferigo Aug 3, 2023
78603bd
Add decorators to apply jit and vmap on jax dataclasses methods
diegoferigo Aug 4, 2023
710e630
Add tests for applying jax transformations in OOP on jax dataclasses
diegoferigo Aug 4, 2023
7b7bafc
Add typing_extension dependency for Python < 3.11
diegoferigo Jul 31, 2023
c7faf4e
Use typing.Self in jaxsim_dataclass
diegoferigo Jul 31, 2023
330131d
Initialize __mutability__ in jaxsim_dataclass
diegoferigo Jul 31, 2023
8829b06
Restore the original pytree content if mutable_context raises
diegoferigo Aug 3, 2023
1111942
Extend the tracing detection checking also the Tracer type
diegoferigo Aug 3, 2023
48c54f0
Run tests in subprocess and move pytest configuration to pyproject.toml
diegoferigo Aug 3, 2023
e82fd08
[wip]
diegoferigo May 18, 2023
741561c
[wip]
diegoferigo May 22, 2023
9be7dc4
[wip]
diegoferigo May 22, 2023
c37bb8f
[wip]
diegoferigo May 22, 2023
0b02457
[models]
diegoferigo May 22, 2023
833e7f0
[sq] jaxgym
diegoferigo Jun 9, 2023
73d1039
[sq] jaxsim.training
diegoferigo Jun 9, 2023
9c2362e
[wip] gitignore
diegoferigo Jun 9, 2023
2f82ed0
[wip] dirty commit with rebase
flferretti Sep 6, 2023
918fdcc
Add training script for ErgoCub environment
flferretti Sep 6, 2023
07ab79a
Fix jax.numpy deprecations
Oct 3, 2023
94dc7ab
Fix JAX deprecation and recursion error
Oct 3, 2023
263fa9e
[wip] working ErgoCub environment
Oct 3, 2023
9387b48
Manage empty arrays when checking spaces
Oct 3, 2023
e6650fb
Clean up and format
flferretti Oct 3, 2023
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 4 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
@@ -1,3 +1,7 @@
core
*.pickle
*.zip

# IDEs
.idea*

Expand Down
16 changes: 16 additions & 0 deletions .vscode/launch.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
{
// Use IntelliSense to learn about possible attributes.
// Hover to view descriptions of existing attributes.
// For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
"version": "0.2.0",
"configurations": [
{
"name": "Python: Current File",
"type": "python",
"request": "launch",
"program": "${file}",
"console": "integratedTerminal",
"justMyCode": true
},
]
}
Loading