torchax is a PyTorch frontend for JAX. It gives JAX the ability to author JAX programs using familiar PyTorch syntax. It also provides JAX-Pytorch interoperability, meaning, one can mix JAX & Pytorch syntax together when authoring ML programs, and run it in every hardware JAX can run.
☆178Feb 21, 2026Updated last week
Alternatives and similar repositories for torchax
Users that are interested in torchax are comparing it to the libraries listed below
Sorting:
- a Jax quantization library☆93Updated this week
- ☆15Jul 13, 2025Updated 7 months ago
- ☆21Feb 3, 2026Updated 3 weeks ago
- Package for data-driven and phenomenological gravitational waveform models☆12Feb 12, 2025Updated last year
- Clean version of research-projects, just with ILE☆18Updated this week
- Detecting compact binary mergers from gravitational wave strain data using neural networks☆18Feb 11, 2026Updated 2 weeks ago
- Measuring Thinking Efficiency in Reasoning Models - Research Repository☆39Dec 2, 2025Updated 2 months ago
- General Relativistic Orbital Dynamics in Julia☆20Dec 12, 2024Updated last year
- Performing parameter estimation on gravitational wave data with machine learning☆22Feb 9, 2026Updated 2 weeks ago
- ☆23Nov 17, 2022Updated 3 years ago
- Prototype routines for GPU quantization written using PyTorch.☆21Feb 8, 2026Updated 3 weeks ago
- ☆22Sep 18, 2023Updated 2 years ago
- Your favourite classical machine learning algos on the GPU/TPU☆21Dec 14, 2025Updated 2 months ago
- Mirror of LALSuite, please use official GitLab repository.☆27Updated this week
- a pipeline for using api calls to agnostically convert unstructured data into structured training data☆32Sep 22, 2024Updated last year
- TORCH_TRACE parser for PT2☆78Feb 12, 2026Updated 2 weeks ago
- Multi-agent simulator in Jax for research and teaching in AI & ALife☆31Updated this week
- Implementation of Action Matching for the Schrödinger equation☆25Jun 18, 2023Updated 2 years ago
- quick playground to animate pippin☆14Nov 11, 2024Updated last year
- Minimal yet performant LLM examples in pure JAX☆240Jan 14, 2026Updated last month
- 🐈 Differentiable Spectrum Modeling of Exoplanets and Brown Dwarfs☆67Jan 27, 2026Updated last month
- Jaxpr Visualisation Tool☆36Dec 22, 2024Updated last year
- Codes and datasets for adaptive spline fitting method SHAPES☆10Sep 27, 2024Updated last year
- Repo to reproduce the First-Explore paper results☆39Dec 25, 2024Updated last year
- ☆270Updated this week
- Orbax provides common checkpointing and persistence utilities for JAX users☆482Updated this week
- FeynGrav is a Wolfram Mathematica package that implements Feynman rules for gravity within FeynCalc framewok.☆12Updated this week
- An open-source package for neutron star whole workflow Bayesian inference constraining Neutron star EOS package☆48Jan 21, 2026Updated last month
- Next Generation Genomic Prediction Tools☆10Dec 3, 2025Updated 2 months ago
- Pseudopotential converter from upf to psp8☆11Jan 25, 2023Updated 3 years ago
- Prior Sampling for high dimension data with domain knowledge.☆10Jan 11, 2022Updated 4 years ago
- JAX accelerated Quantum Monte Carlo☆92Aug 11, 2025Updated 6 months ago
- Next generation Bayesian analysis tools for efficient posterior sampling and evidence estimation.☆27Oct 13, 2022Updated 3 years ago
- Library of Octave functions for continuous gravitational-wave data analysis☆12Feb 28, 2023Updated 3 years ago
- ✈️🏞️ Flying airplane in natural terrain generated by Perlin Noise.☆10Nov 20, 2022Updated 3 years ago
- Gravitational wave interferometer parameter optimisation game, written in Python and run in a Jupyter notebook.☆10Dec 18, 2018Updated 7 years ago
- Triton-based Symmetric Memory operators and examples☆85Jan 15, 2026Updated last month
- Disable YubiKey output on MacOS without a modifier key pressed☆10Aug 10, 2022Updated 3 years ago
- Discord Docsbot, Built on bgent☆11Jun 17, 2024Updated last year