Toy implementations of some popular ML optimizers using Python/JAX
☆44Jun 20, 2021Updated 5 years ago
Alternatives and similar repositories for ML-Optimizers-JAX
Users that are interested in ML-Optimizers-JAX are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Paper introducing jax-cosmo☆13Apr 27, 2023Updated 3 years ago
- An approximate implementation of the OpenAI paper - An Empirical Model of Large-Batch Training for MNIST☆11Nov 19, 2022Updated 3 years ago
- Reinforcement Learning workshop☆18Feb 11, 2020Updated 6 years ago
- Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation☆12May 24, 2021Updated 5 years ago
- Source code of "What can linearized neural networks actually say about generalization?☆20Oct 21, 2021Updated 4 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Yet Another Neural Network Library 🤔☆27Feb 23, 2026Updated 4 months ago
- Annotations of the interesting ML papers I read☆286Jun 6, 2026Updated last month
- ☆13Jul 25, 2024Updated last year
- Using FlexAttention to compute attention with different masking patterns☆47Sep 22, 2024Updated last year
- PyTorch implementation of Lessons on Parameter Sharing across Layers in Transformers☆27May 19, 2021Updated 5 years ago
- Implementation of numerous Vision Transformers in Google's JAX and Flax.☆22Aug 30, 2022Updated 3 years ago
- Machine Reading Comprehension has attracted significant interest in research on natural language understanding, and large-scale datasets …☆10Aug 14, 2021Updated 4 years ago
- ☆15Sep 13, 2022Updated 3 years ago
- ☆10Dec 17, 2020Updated 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 re-implementation of the "Red Teaming Language Models with Language Models" paper by Perez et al., 2022☆34Oct 9, 2023Updated 2 years ago
- A re-implementation of the "Extracting Training Data from Large Language Models" paper by Carlini et al., 2020☆39Jul 10, 2022Updated 3 years ago
- A PyTorch Implementation of the Skipgram Negative Sampling Word2Vec Model as Described in Mikolov et al.☆15Jan 13, 2020Updated 6 years ago
- A pathway and collection of resources to learning Jax from beginning to advance.☆11Jan 2, 2021Updated 5 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
- An implementation of the base GPT-3 Model architecture from the paper by OPENAI "Language Models are Few-Shot Learners"☆22Jun 29, 2024Updated 2 years ago
- Personal Machine Learning Projects☆15Dec 23, 2017Updated 8 years ago
- Shape Matching Virtual Element Method for direct simulation on CAD models!☆13Feb 12, 2022Updated 4 years ago
- NoiseGrad (and its extension NoiseGrad++) is a method to enhance explanations of artificial neural networks by adding noise to model weig…☆22May 11, 2023Updated 3 years ago
- End-to-end encrypted email - Proton Mail • AdSpecial offer: 40% Off Yearly / 80% Off First Month. All Proton services are open source and independently audited for security.
- Code for the Paper 'On the Connection Between Adversarial Robustness and Saliency Map Interpretability' by C. Etmann, S. Lunz, P. Maass, …☆16May 9, 2019Updated 7 years ago
- ☆11Oct 22, 2020Updated 5 years ago
- A pytorch implementation of https://arxiv.org/abs/1903.01434☆12Jan 21, 2020Updated 6 years ago
- Meeting repo for likelihood free inference meeting☆14Oct 6, 2022Updated 3 years ago
- 📈Implementing the ADAM optimizer from the ground up with PyTorch and comparing its performance on six 3-D objective functions (each prog…☆23Jul 2, 2022Updated 4 years ago
- CNN for Multiple Choice Question Answering☆19Feb 21, 2020Updated 6 years ago
- A JAX implementation of stochastic addition.☆14Aug 15, 2022Updated 3 years ago
- MNIST, but with Bezier curves instead of pixels☆15Oct 29, 2021Updated 4 years ago
- Simple, extensible implementations of some meta-learning algorithms in Jax☆11Oct 6, 2020Updated 5 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Code of ICML paper arxiv.org/abs/2302.08105☆14May 4, 2023Updated 3 years ago
- ☆12Sep 2, 2021Updated 4 years ago
- About Question Difficulty Prediction Based on Neural Network.☆39Nov 18, 2020Updated 5 years ago
- List of awesome JAX resources☆13Dec 8, 2022Updated 3 years ago
- Resources on Reinforcement Learning☆14Aug 9, 2018Updated 7 years ago
- ☆17Jun 14, 2026Updated 3 weeks ago
- For visual commonsense model☆34Apr 12, 2019Updated 7 years ago