basicv8vc / jax-tutorials-for-pytorchersLinks
JAX tutorials for PyTorch users
☆13Updated 2 years ago
Alternatives and similar repositories for jax-tutorials-for-pytorchers
Users that are interested in jax-tutorials-for-pytorchers are comparing it to the libraries listed below
Sorting:
- [ICML 2022] NeuroFluid: Fluid Dynamics Grounding with Particle-Driven Neural Radiance Fields☆88Updated 2 years ago
- numerical optimization in pytorch☆139Updated last year
- ☆12Updated 3 months ago
- Constrained Diffusion with Trust Sampling☆12Updated last year
- [ICML 2020] Visual Grounding of Learned Physical Models (Dynamics Prior)☆40Updated last year
- ☆31Updated 3 years ago
- Code for "Accurate Differential Operators for Hybrid Neural Fields", accepted at CVPR 2025☆28Updated 8 months ago
- [ECCV '22] Intrinsic Neural Fields: Learning Functions on Manifolds☆77Updated 3 years ago
- Implementation of a Trust Region Newton Conjugate Gradient optimizer in Pytorch☆36Updated 3 years ago
- ☆23Updated 2 years ago
- Trading Positional Complexity vs Deepness in Coordinate Networks☆31Updated 2 years ago
- Code for paper "Implicit Neural Spatial Representations for Time-dependent PDEs", ICML 2023☆46Updated 2 years ago
- Python implementation of the Levenberg-Marquardt algorithm☆32Updated 4 years ago
- Official implementation of Generative Cellular Automata: ICLR 2021, 2022 (spotlight)☆54Updated 9 months ago
- A Transformer made of Rotation-equivariant Attention using Vector Neurons☆101Updated 2 years ago
- ☆21Updated last year
- [ECCV 2022] The official experimental code of "Sobolev Training for Implicit Neural Representations with Approximated Image Derivatives"☆32Updated 3 years ago
- [CVPR 2022] Projective Manifold Gradient Layer For Deep Rotation Regression☆74Updated 9 months ago
- Code accompanying the NeurIPS 2022 paper: "NeuPhysics: Editable Neural Geometry and Physics from Monocular Videos".☆153Updated 2 years ago
- Official code for Inferring Hybrid Neural Fluid Fields from Videos (NeurIPS 2023)☆59Updated 2 years ago
- Public Repo for the paper "Overcoming The Spectral-Bias of Neural Value Approximation"☆11Updated last year
- Official JAX implementation of neural isometries - taming transformations for equivariant ML☆36Updated 6 months ago
- Polynomial Neural Fields for Subband Decomposition and Manipulation☆58Updated 2 years ago
- Geometry processing utilities compatible with jax for autodifferentiation.☆90Updated 2 years ago
- Contact points matter for squishing play-doh into an airplane!☆39Updated 3 years ago
- PyTorch implementation of the Hessian-free optimizer☆36Updated last year
- Implicit^2: Implicit model for implicit neural representations☆29Updated 4 years ago
- Official code for the paper DiffSkill: Skill Abstraction from Differentiable Physics for Deformable Object Manipulations with Tools☆33Updated 2 years ago
- [ICML 2023] Learning Neural Constitutive Laws from Motion Observations for Generalizable PDE Dynamics☆146Updated 2 years ago
- Code for the RSS 2023 paper "Energy-based Models are Zero-Shot Planners for Compositional Scene Rearrangement"☆21Updated 2 years ago