jaxgptoolbox
jaxgptoolbox copied to clipboard
Geometry processing utilities compatible with jax for autodifferentiation.
Hi, first, I love the project. I was wondering if you might have working code for the PointNet chair example for LipMLP, somewhere, that users can easily try?
To facilitate a more efficient installation process, could you please provide a 'requirements.txt' file for pip or a 'requirements.yml' file for conda, if available? Your assistance is highly appreciated. Best...
Dear authors, Thanks for releasing the code! I am wondering if it is possible to also provide the code in pytorch? I want to compare the results with DeepSDF. Thanks...
Hi, congrats on your LipMLP work! According to the paper appendix A, the weight matrix should be initialized with the He init if ReLU is used. Regarding the code (https://github.com/ml-for-gp/jaxgptoolbox/blob/main/demos/lipschitz_mlp/model.py#L33)...