samuela / torch2jaxLinks
Run PyTorch in JAX. 🤝
☆292Updated 3 weeks ago
Alternatives and similar repositories for torch2jax
Users that are interested in torch2jax are comparing it to the libraries listed below
Sorting:
- Pytorch-like dataloaders for JAX.☆94Updated 4 months ago
- ☆217Updated this week
- JAX Arrays for human consumption☆106Updated 2 months ago
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆470Updated 3 weeks ago
- Use Jax functions in Pytorch