Simple CIFAR10 ResNet example with JAX.
☆23Jun 1, 2021Updated 5 years ago
Alternatives and similar repositories for JAX-ResNet-CIFAR10
Users that are interested in JAX-ResNet-CIFAR10 are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- This is the official implementation of the ICML 2023 paper - Can Forward Gradient Match Backpropagation ?☆13May 31, 2023Updated 3 years ago
- ☆15Dec 16, 2025Updated 7 months ago
- Derivative-Informed Neural Operator: An Efficient Framework for High-Dimensional Parametric Derivative Learning☆18Jan 9, 2024Updated 2 years ago
- An engine for constructing PDE-based simulations using Firedrake☆21Oct 24, 2023Updated 2 years ago
- ☆38Jan 8, 2026Updated 6 months ago
- Simple, predictable pricing with DigitalOcean hosting • AdAlways know what you'll pay with monthly caps and flat pricing. Enterprise-grade infrastructure trusted by 600k+ customers.
- Keras implementation of m2det object detection.☆48Aug 21, 2019Updated 6 years ago
- ☆10Aug 18, 2016Updated 9 years ago
- Experiments on the impact of depth in transformers and SSMs.☆44Oct 23, 2025Updated 9 months ago
- ☆16Dec 29, 2025Updated 7 months ago
- ☆16Dec 29, 2025Updated 7 months ago
- Code for Augment & Reduce, a scalable stochastic algorithm for large categorical distributions☆10May 16, 2018Updated 8 years ago
- ☆35Feb 10, 2025Updated last year
- ☆11Sep 1, 2024Updated last year
- BESA is a differentiable weight pruning technique for large language models.☆17Mar 4, 2024Updated 2 years ago
- 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.
- An extension of Fourier Neural Operator to finite-dimensional input and/or output spaces.☆19Oct 13, 2025Updated 9 months ago
- Group project "Algorithms for large-scale optimal transport". Implement ADMMs and Sinkhorn's Algorithms.☆11Jan 28, 2019Updated 7 years ago
- codes and plots for "Active-Dormant Attention Heads: Mechanistically Demystifying Extreme-Token Phenomena in LLMs"☆11Dec 30, 2024Updated last year
- Implementation of the paper 'Stochastic Wasserstein Barycenters'☆11Oct 17, 2018Updated 7 years ago
- ☆10Aug 26, 2022Updated 3 years ago
- Jupyter notebooks from our weekly (or so) hackathons☆11Dec 3, 2024Updated last year
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆47Mar 24, 2023Updated 3 years ago
- Official Pytorch implementation of the ECCV 2022 paper "GIPSO: Geometrically Informed Propagation for Online Adaptation in 3D LiDAR Segme…☆49Jan 13, 2023Updated 3 years ago
- Large-batch Training, Neural Network Optimization☆10Nov 8, 2019Updated 6 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- A GPT, made only of MLPs, in Jax☆59Jun 23, 2021Updated 5 years ago
- Atmospheric Transport Model in pure Julia - GPU-CPU capable, ERA or GEOS met sources, LatLon, Reduced Gaussian and CubeSphere Topologies☆23Jul 19, 2026Updated last week
- Python and MATLAB code for Stein Variational sampling methods☆28May 25, 2019Updated 7 years ago
- ☆16Apr 26, 2023Updated 3 years ago
- Toolbox for Unsupervised Anomaly Detection on MVTec AD☆11Apr 23, 2021Updated 5 years ago
- This is a public repository for:☆38Aug 11, 2021Updated 4 years ago
- A two-level sparse direct solver for elliptic PDEs☆31Apr 3, 2026Updated 3 months ago
- ☆16Nov 20, 2023Updated 2 years ago
- CIFAR-5m dataset☆41Dec 28, 2020Updated 5 years ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- [COLM 2026] An efficient 3D sampling method for long-CoT LLM.☆16May 25, 2025Updated last year
- Posterior with interesting shapes from actually used models☆13Feb 10, 2025Updated last year
- Fast interpolative decompositions in Python☆10Jan 4, 2021Updated 5 years ago
- [Findings of ACL 2023] Communication Efficient Federated Learning for Multilingual Machine Translation with Adapter☆12Sep 4, 2023Updated 2 years ago
- ☆13Mar 10, 2026Updated 4 months ago
- This repo is the official implementation for the series of works on (Path-dependent) Neural Jump ODEs.☆27Feb 8, 2026Updated 5 months ago
- [ICLR 2025] RaSA: Rank-Sharing Low-Rank Adaptation☆10May 19, 2025Updated last year