Running Jax in PyTorch Lightning
☆122Dec 16, 2024Updated last year
Alternatives and similar repositories for JaxLightning
Users that are interested in JaxLightning are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Lightning-like training API for JAX with Flax☆45Dec 8, 2024Updated last year
- A trivial CLI library. https://docs.kidger.site/seali/☆13May 10, 2026Updated 5 months ago
- ☆26Feb 15, 2023Updated 3 years ago
- ☆45Nov 18, 2022Updated 3 years ago
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/☆2,999Oct 25, 2026Updated 2 weeks ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- ☆11Jun 3, 2024Updated 2 years ago
- Agent-assisted hyperparameter optimization 📈☆20Aug 2, 2026Updated 3 months ago
- Code to reproduce experiments in Markovian Flow Matching: Accelerating MCMC with Continuous Normalizing Flows☆14May 23, 2024Updated 2 years ago
- PLaTITO is an open-source model for generative molecular dynamics of proteins.☆28Oct 12, 2026Updated 3 weeks ago
- An implementation of SchNet in JAX and JAX-MD.☆17Apr 5, 2022Updated 4 years ago
- Template for writing reproducible machine learning papers☆13May 18, 2023Updated 3 years ago
- Lightning HPO & Training Studio App☆18Apr 12, 2023Updated 3 years ago
- Reference implementation of "Ewald-based Long-Range Message Passing for Molecular Graphs" (ICML 2023)☆52May 4, 2026Updated 6 months ago
- Rubric compiler and judge engine for LLM evaluation☆15Jun 20, 2026Updated 4 months ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- A set of utilities to turn Dataclasses into useful configuration managers.☆11Mar 27, 2024Updated 2 years ago
- Perceiver (transformer variant) implemented in JAX and Flax☆13Mar 29, 2021Updated 5 years ago
- "Not too complicated" training code for CIFAR-10 by PyTorch Lightning☆12Jun 5, 2022Updated 4 years ago
- Data and model repository for the ASKCOS application☆16Feb 24, 2022Updated 4 years ago
- Code for Discrete Beckmann Transport Models for One-Step Language Modeling and Reasoning☆13Sep 15, 2026Updated last month
- ☆25Nov 1, 2023Updated 3 years ago
- ☆16Dec 26, 2018Updated 7 years ago
- Vectorization techniques for fast population-based training.☆58Apr 26, 2026Updated 6 months ago
- A library implementing the kernels for and experiments using extrinsic gauge equivariant vector field Gaussian Processes☆26Oct 28, 2021Updated 5 years ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- A small library of paramaterizations and parameter constraints for PyTrees.☆68Jan 23, 2026Updated 9 months ago
- Code for the paper: "Generalized Discrete Diffusion from Snapshots"☆14Apr 2, 2026Updated 7 months ago
- CIFAR10 ResNets implemented in JAX+Flax☆12Apr 6, 2022Updated 4 years ago
- ☆12Jan 9, 2024Updated 2 years ago
- ☆16May 30, 2023Updated 3 years ago
- Gradient-informed particle MCMC methods☆12Jan 29, 2024Updated 2 years ago
- Coordinate Axes for JAX☆67Oct 12, 2026Updated 3 weeks ago
- Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/☆1,881Oct 25, 2026Updated 2 weeks ago
- PyTorch Lightning template for rapid, scalable and reproducible ML experimentation with best practices.☆24Oct 8, 2025Updated last year
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- A pytorch toolbox designed for experimentation using a generalization of the Resnet design.☆15Jul 6, 2023Updated 3 years ago
- Research Project Template Repository☆54May 6, 2026Updated 6 months ago
- Maximal Update Parametrization (μP) with Flax & Optax.☆16Dec 27, 2023Updated 2 years ago
- QDrant-NLP☆12Aug 30, 2022Updated 4 years ago
- ☆53Updated this week
- Simple repository contribution statistics☆15Nov 1, 2026Updated last week
- Quantifying degeneracy in toy models☆11Sep 6, 2024Updated 2 years ago
Are these results useful?