About this role
Vynexa seeks an AI & ML Researcher to develop applied AI models and systems at the intersection of research and production. The full-time role is listed as Chandigarh / Remote.
Responsibilities
- Read and implement recent research on architectures including Transformers, Diffusion and Mamba.
- Train and fine-tune custom small language models for domain-specific tasks.
- Optimize inference pipelines for speed and efficiency on consumer hardware.
- Experiment with novel architectures and turn research into production product features.
Requirements
- Deep understanding of how large language models and Diffusion models work, beyond using their APIs.
- Fluency in PyTorch and experience optimizing models through quantization, pruning, distillation, ONNX or TensorRT.
- Ability to implement research papers in code within days, strong software engineering fundamentals, and a portfolio demonstrating AI projects that work in practice.
Preferred
Published research or a significant open-source presence in the AI community. The posting mentions free food and some energy drinks in the company's apartment; it does not state a salary.
Skills for this role
PyTorchTransformersDiffusion modelsMambaLarge language modelsSmall language modelsModel trainingFine-tuningInference optimizationQuantizationPruningDistillationONNXTensorRTCUDAResearch implementationSoftware engineering