Pytorch2Jax is a small Python library that provides functions that wraps PyTorch models into Jax functions and Flax modules.
☆21Feb 20, 2023Updated 3 years ago
Alternatives and similar repositories for pytorch2jax
Users that are interested in pytorch2jax are comparing it to the libraries listed below
Sorting:
- [AAMAS 2023] Code for the paper "Automatic Noise Filtering with Dynamic Sparse Training in Deep Reinforcement Learning"☆12Feb 22, 2024Updated 2 years ago
- Implementation of Proximal Policy Optimization in Jax+Flax☆21May 18, 2023Updated 2 years ago
- Docker containers of baseline agents for the Crafter environment☆30Dec 14, 2021Updated 4 years ago
- Differentiable models of halo and galaxy formation history☆10Updated this week
- Materials for MC-Toolkit meetings at IGE☆12Jul 6, 2023Updated 2 years ago
- Weak lensing measurement pipeline for wide area surveys☆14Apr 14, 2022Updated 3 years ago
- A Nextflow pipeline that creates reliable, structure-informed MSAs of thousands of protein sequences which can supplement structural info…☆11Nov 19, 2025Updated 3 months ago
- Real-Time RTUs☆11Jan 2, 2025Updated last year
- The repo for paper: Exploiting the Index Gradients for Optimization-Based Jailbreaking on Large Language Models.☆13Dec 16, 2024Updated last year
- Synthetic galaxy cluster images and catalogs for optical sky surveys☆15Sep 28, 2023Updated 2 years ago
- Vacances scolaires en France☆16Nov 14, 2025Updated 3 months ago
- ☆13Updated this week
- ☆11Nov 13, 2024Updated last year
- Deep Multi-Agent Reinforcement Learning with StarCraft 2☆10Sep 27, 2020Updated 5 years ago
- MMD-FUSE package implementing the MMD-FUSE test proposed in MMD-FUSE: Learning and Combining Kernels for Two-Sample Testing Without Data …☆11May 31, 2024Updated last year
- Dockerfile for serving bokeh visualisations☆11Oct 6, 2020Updated 5 years ago
- Benchmarking field-level cosmological inference from galaxy surveys.☆12Jul 17, 2025Updated 7 months ago
- a differentiable lensing simulator☆10Jan 18, 2022Updated 4 years ago
- 🔦 A minimal raytracing engine in written in C on MinilibX☆10Mar 23, 2021Updated 4 years ago
- tutorials on shape measurement☆10Jan 5, 2022Updated 4 years ago
- Experiment utility code, specifically designed for use with Compute Canada.☆11Jan 27, 2025Updated last year
- RealSim is the statistical observational realism suite described in Bottrell et al 2017ab and made public in Bottrell et al 2019b.☆12Jul 7, 2021Updated 4 years ago
- Analysis workflow for investigating the structural context of PTMs☆11Jan 8, 2024Updated 2 years ago
- A Python wrapper for the lightning fast fast3tree library, a BSP tree implementation written by Peter Behroozi.☆13Jan 12, 2026Updated last month
- A conda-smithy repository for nvcc.☆13Jan 23, 2025Updated last year
- Implicit neural representations for strong lensing source reconstruction. Code repository associated with https://arxiv.org/abs/2206.1482…☆12Feb 26, 2023Updated 3 years ago
- This is the code for "Crypto Math" by Siraj Raval on Youtube☆28Jan 24, 2018Updated 8 years ago
- A code for generating fake spectra from a cosmological simulation☆16Aug 10, 2025Updated 6 months ago
- Exploring how ChatGPT can be used to accelerate research in cosmology.☆12Dec 12, 2022Updated 3 years ago
- Investigating the information content in the cosmic web through dark matter halo graphs. Arxiv paper: https://arxiv.org/abs/2207.05202☆13Jan 18, 2024Updated 2 years ago
- sentinel 3 optical playground☆13Apr 18, 2017Updated 8 years ago
- generate realistic atmospheric input parameters for point-spread function simulations in GalSim☆16Oct 29, 2024Updated last year
- Autoencoders for galaxies☆11Jul 29, 2021Updated 4 years ago
- An open-source platform for developing protein models beyond AlphaFold.☆12Jul 15, 2025Updated 7 months ago
- E(n) Equivariant GNN in jax☆14Aug 31, 2023Updated 2 years ago
- (Official) PyTorch implementation for Efficient Episodic Memory Utilization of Cooperative Multi-Agent Reinforcement Learning (EMU) in IC…☆12May 23, 2024Updated last year
- A TensorFlow reimplementation of GalSim☆10Mar 8, 2022Updated 3 years ago
- Multi-object/band source modelling/galaxy fitting code☆11Nov 6, 2024Updated last year
- Cloud enabled snippet manager on terminal.☆39Jan 8, 2018Updated 8 years ago