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:
- A Transformer made of Rotation-equivariant Attention using Vector Neurons☆99Updated 2 years ago
- a thin wrapper of chatgpt for improving paper writing.☆253Updated 2 years ago
- Python implementation of the Levenberg-Marquardt algorithm☆32Updated 4 years ago
- ☆12Updated 2 months ago
- PyTorch implementation of the Hessian-free optimizer☆36Updated last year
- numerical optimization in pytorch☆139Updated last year
- Official implementation of Generative Cellular Automata: ICLR 2021, 2022 (spotlight)☆52Updated 8 months ago
- Code for "Accurate Differential Operators for Hybrid Neural Fields", accepted at CVPR 2025☆28Updated 7 months ago
- [ICML 2022] NeuroFluid: Fluid Dynamics Grounding with Particle-Driven Neural Radiance Fields☆88Updated 2 years ago
- [ECCV '22] Intrinsic Neural Fields: Learning Functions on Manifolds☆77Updated 3 years ago
- ☆35Updated 4 years ago
- Mirror Diffusion Models, NeurIPS 2023☆34Updated 2 years ago
- LieTransformer: Equivariant Self-Attention for Lie Groups☆77Updated last year
- [ECCV 2022] The official experimental code of "Sobolev Training for Implicit Neural Representations with Approximated Image Derivatives"☆32Updated 3 years ago
- Cubic spline interpolation on multidimensional grids in PyTorch☆34Updated last month
- Research project for real-time rendering using Neural Radiance Fields (NeRF) and invertible neural networks (INNs)☆14Updated 4 years ago
- Equivariant Point Cloud Analysis via Learning Orientations for Message Passing (CVPR 2022)☆29Updated 3 years ago
- Implicit^2: Implicit model for implicit neural representations☆29Updated 4 years ago
- Method to predict distributions over 3D object rotation from 2D images☆35Updated 2 years ago
- Contact points matter for squishing play-doh into an airplane!☆39Updated 3 years ago
- Source code for Fathony, Sahu, Willmott, & Kolter, "Multiplicative Filter Networks", ICLR 2021.☆96Updated 4 years ago
- Polynomial Neural Fields for Subband Decomposition and Manipulation☆58Updated 2 years ago
- ☆36Updated 2 years ago
- Zhusuan with backend Jittor☆15Updated 3 years ago
- Geometry processing utilities compatible with jax for autodifferentiation.☆90Updated 2 years ago
- [ICLR'24] GTA: A Geometry-Aware Attention Mechanism for Multi-view Transformers