IDRIS-CNRS / DLO-JZLinks
formation Deep Learning Optimisé pour Jean Zay
☆17Updated 2 months ago
Alternatives and similar repositories for DLO-JZ
Users that are interested in DLO-JZ are comparing it to the libraries listed below
Sorting:
- ☆367Updated 3 months ago
- TUI for the Slurm Workload Manager☆394Updated 2 weeks ago
- Compositional Linear Algebra☆504Updated 5 months ago
- CLU lets you write beautiful training loops in JAX.☆363Updated 3 weeks ago
- Running Jax in PyTorch Lightning☆118Updated last year
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.☆303Updated 3 weeks ago
- Run PyTorch in JAX. 🤝☆309Updated 2 months ago
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆508Updated 3 weeks ago
- ☆261Updated 2 weeks ago
- JMP is a Mixed Precision library for JAX.☆210Updated 11 months ago
- JAX Arrays for human consumption☆110Updated 2 months ago
- SC24 Deep Learning at Scale Tutorial Material☆33Updated 11 months ago
- Material for the SC22 Deep Learning at Scale Tutorial☆41Updated 2 years ago
- Agustinus' very opiniated publication-ready plotting library☆70Updated 8 months ago
- Orbax provides common checkpointing and persistence utilities for JAX users☆474Updated this week
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆406Updated this week
- ☆233Updated 10 months ago
- JAX Synergistic Memory Inspector☆183Updated last year
- Example of how to use Weights & Biases on Slurm☆118Updated 3 years ago
- jax-triton contains integrations between JAX and OpenAI Triton☆436Updated 3 weeks ago
- A dummy's guide to setting up (and using) HPC clusters on Ubuntu 22.04LTS using Slurm and Munge. Created by the Quant Club @ UIowa.☆399Updated last year
- TensorDict is a pytorch dedicated tensor container.☆998Updated last week
- Minimal yet performant LLM examples in pure JAX☆225Updated last week
- JAX-Toolbox☆373Updated this week
- The LUMI AI Guide is designed to assist users in migrating their machine learning applications from smaller-scale computing environments …☆61Updated last month
- Universal, autodiff-native software components for Simulation Intelligence. 📦☆85Updated this week
- A Python package of computer vision models for the Equinox ecosystem.☆109Updated last year
- Collaborative documentation for and from Jean Zay users. Official Jean Zay documentation: http://www.idris.fr/eng/jean-zay/☆128Updated last month
- Library for reading and processing ML training data.☆643Updated last week
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆688Updated 2 weeks ago