☆24Jul 31, 2023Updated 2 years ago
Alternatives and similar repositories for jax-gravity
Users that are interested in jax-gravity are comparing it to the libraries listed below
Sorting:
- A functional training loops library for JAX☆88Feb 13, 2024Updated 2 years ago
- This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.☆14Dec 21, 2021Updated 4 years ago
- ☆13Jul 12, 2024Updated last year
- Transformers training in a supercomputer with the 🤗 Stack and Slurm☆15May 9, 2024Updated last year
- Colab Notebooks for JAX/Flax/Optax ML Book☆12Feb 25, 2023Updated 3 years ago
- A metrics library for the JAX ecosystem☆40Mar 16, 2023Updated 2 years ago
- A collection of reusable, high-performance, well-documented, thorough-tested layers and models in Jax☆23Jun 8, 2025Updated 8 months ago
- Code for the paper "Tensor Networks for Maching Learning"☆17Nov 7, 2019Updated 6 years ago
- FID computation in Jax/Flax.☆29Jul 17, 2024Updated last year
- How to use the Flax Linen API to build a convolutional neural network model and train it for image classification (using TensorFlow Datas…☆25Aug 16, 2023Updated 2 years ago
- A simple script to recompile arxiv papers into kindle-like format☆29Oct 4, 2023Updated 2 years ago
- This repository hosts code for converting the original Vision Transformer models (JAX) to TensorFlow.☆33Mar 23, 2022Updated 3 years ago
- JAX Implementation of Black Forest Labs' Flux.1 family of models☆40Feb 9, 2026Updated 3 weeks ago
- A differentiation API for PyTorch☆30Jun 21, 2020Updated 5 years ago
- R Shiny application "Vaccine Designer" aiming for the construction of vaccine sequences based on multi epitope design workflow.☆10Jun 25, 2024Updated last year
- Normalizing Flows using JAX☆86Nov 30, 2023Updated 2 years ago
- PSI-MOD ontology for modified and unmodified amino acid residues☆14Jan 8, 2026Updated last month
- C extension for Python, calculates the permanent☆10Oct 28, 2024Updated last year
- TensorFlow 2 / Lite implementation of Ultra-Fast Structure-Aware Lane Detection☆12Aug 19, 2020Updated 5 years ago
- Canonical normalizing flows☆10Apr 30, 2019Updated 6 years ago
- Simple tutorial to get familiar with how to program quantum computers using Qiskit☆11Sep 9, 2019Updated 6 years ago
- Supporting code for Deep Neural Network or Dermatologist?☆10Nov 14, 2019Updated 6 years ago
- ☆12Feb 6, 2026Updated 3 weeks ago
- PyTorch and TensorFlow/Keras image models with automatic weight conversions and equal API/implementations - Vision Transformer (ViT), Res…☆42Jul 1, 2023Updated 2 years ago
- introduction to dataflow analysis using julia☆14Oct 26, 2020Updated 5 years ago
- A quick realization of the quantum circuit of a matrix product state.☆11Aug 23, 2023Updated 2 years ago
- Batched routines (BLAS, LAPACK, etc.) for multi-dimensional arrays☆12Apr 10, 2022Updated 3 years ago
- Vehicle control☆11Jun 8, 2019Updated 6 years ago
- Prediction of glycopeptide fragment mass spectra by deep learning☆10Feb 20, 2024Updated 2 years ago
- ☆11Mar 13, 2021Updated 4 years ago
- TPS Tutorial FOR1807☆10Feb 22, 2018Updated 8 years ago
- EuBIC-MS Developers Meeting 2025 on Computational Mass Spectrometry☆11Nov 8, 2024Updated last year
- Implementation of Variance Reduction Techniques in Julia☆11Sep 6, 2016Updated 9 years ago
- Groundstate Reconstructive Energies using Neural Autoregressive Distribution Estimators (GreNADE)☆10Apr 1, 2022Updated 3 years ago
- Extension of almaBTE's Monte Carlo code beyond the relaxation-time approximation.☆10Sep 24, 2023Updated 2 years ago
- ☆20Feb 13, 2026Updated 2 weeks ago
- Code for the paper "Variational Monte Carlo Approach to Partial Differential Equations with Neural Networks" (https://arxiv.org/abs/2206.…☆10Aug 8, 2022Updated 3 years ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆46Jan 11, 2025Updated last year
- ☆12Feb 15, 2026Updated 2 weeks ago